CodeForces Round #515 Div.3 B. Heaters
http://codeforces.com/contest/1066/problem/B
Vova's house is an array consisting of nn elements (yeah, this is the first problem, I think, where someone lives in the array). There are heaters in some positions of the array. The ii-th element of the array is 11 if there is a heater in the position ii, otherwise the ii-th element of the array is 00.
Each heater has a value rr (rr is the same for all heaters). This value means that the heater at the position pospos can warm up all the elements in range [pos−r+1;pos+r−1][pos−r+1;pos+r−1].
Vova likes to walk through his house while he thinks, and he hates cold positions of his house. Vova wants to switch some of his heaters on in such a way that each element of his house will be warmed up by at least one heater.
Vova's target is to warm up the whole house (all the elements of the array), i.e. if n=6n=6, r=2r=2 and heaters are at positions 22 and 55, then Vova can warm up the whole house if he switches all the heaters in the house on (then the first 33 elements will be warmed up by the first heater and the last 33 elements will be warmed up by the second heater).
Initially, all the heaters are off.
But from the other hand, Vova didn't like to pay much for the electricity. So he wants to switch the minimum number of heaters on in such a way that each element of his house is warmed up by at least one heater.
Your task is to find this number of heaters or say that it is impossible to warm up the whole house.
The first line of the input contains two integers nn and rr (1≤n,r≤10001≤n,r≤1000) — the number of elements in the array and the value of heaters.
The second line contains nn integers a1,a2,…,ana1,a2,…,an (0≤ai≤10≤ai≤1) — the Vova's house description.
Print one integer — the minimum number of heaters needed to warm up the whole house or -1 if it is impossible to do it.
6 2
0 1 1 0 0 1
3
5 3
1 0 0 0 1
2
5 10
0 0 0 0 0
-1
10 3
0 0 1 1 0 1 0 0 0 1
3
代码:
#include <bits/stdc++.h> using namespace std; int N, R; int pos[1010]; int main() { scanf("%d%d", &N, &R); for(int i = 0; i < N; i ++) scanf("%d", &pos[i]); vector<int> T; int st = -1; for(int i = N - 1; i >= 0; i --) { if(pos[i] == 1 && i < R) { st = i; break; } } if(st == -1) { printf("-1\n"); return 0; } if(N - st <= R) { printf("1\n"); return 0; } T.push_back(st); while(true) { // cout << "!!!" << endl; bool flag = false; for(int i = N - 1; i >= st + 1; i --) { if(i - st + 1 <= 2 * R && pos[i]) { st = i; flag = true; break; } } if(!flag) { break; } T.push_back(st); if(N - st <= R) { printf("%d\n", T.size()); return 0; } } printf("-1\n"); return 0; }