adcoder 161 E

題目:https://atcoder.jp/contests/abc161/tasks/abc161_e

技巧性很強。利用貪心的算法,先從左往右遍歷,找到第x(x<k)天最早開始工作的下標i,即>=i。然後從右往左遍歷,找到第x(x<k)最遲開始工作的下標i,即<=i。當最早和最遲兩者取等的時候,這天就沒有其他選擇,必須工作。

#include<iostream>
#include <algorithm>
#include<queue>
using namespace std;


char s[200005];
int n, k, c;
int main() 
{
	cin >> n >> k >> c >> s;
	vector<int> l, r;
	int temp = 0x3f3f3f3f;
	for (int i = 0; i < n; i++)
	{
		if (l.size() == k)
			break;
		if (s[i] == 'o' && temp >= c)
		{
			l.push_back(i);
			temp = 0;
		}
		else
			temp++;
	}
	temp = 0x3f3f3f3f;
	for (int i = n - 1; i >= 0; i--)
	{
		if (r.size() == k)
			break;
		if (s[i] == 'o' && temp >= c)
		{
			r.push_back(i);
			temp = 0;
		}
		else
			temp++;
	}
  	reverse(r.begin(),r.end());
	for (int i = 0; i < k; i++)
	{
		if (l[i] == r[i])
			cout << l[i]+1 << endl;
	}
	return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章