Codeforces Round #380 (Div. 2) D. Sea Battle (贪心)

题目链接:http://codeforces.com/contest/738/problem/D


题意:看样例吧

13 3 2 3
1000000010001
一个游戏,给长度为13的字符串,0表示没炸过,1表示炸过,然后有2个船,每个船长2,炸过3次了(就是有3个1)。

问最少炸多少次能至少保证炸到一艘船?输出该炸的位置。


贪心,代码写的不太优雅。先预处理出了1的位置,然后可以知道每个区间的开始和结束,然后贪心地向后看就行了。

最后ans里得到的是把所有的船全炸掉最少需要多少次,只用输出前size-k+1个就是至少炸掉一艘船的答案了。

#include 
using namespace std;
int n, m;
vector pos, ans;
int main() {
	int n, a, b, k;
	cin >> n >> a >> b >> k;
	string str;
	cin >> str;
	for(int i = 0; i < n; i++) {
		if(str[i]=='1') {
			pos.push_back(i);
		}
	}
	pos.push_back(n);
	int s = 0, e;
	for(int i = 0; i < pos.size(); i++) {
		e = pos[i];
		if(e - s >= b) {
			int cnt = 0;
			for(int j = s; j < e; j++) {
				cnt++;
				if(cnt==b) {
					ans.push_back(j);
					cnt=0;
					if(e - j - 1 < b) break;
				}
			}
		}
		s = e+1;
	}
	cout << ans.size() - a + 1 << endl;
	for(int i = 0; i <= ans.size() - a; i++) {
		if(i == 0) cout << ans[i]+1;
		else cout << ' ' << ans[i]+1;
	}
	cout << endl;
	return 0;
}



你可能感兴趣的:(贪心)