-
Notifications
You must be signed in to change notification settings - Fork 3
/
Maximize Number of 1's By Flipping M zeroes.cpp
73 lines (69 loc) · 2.07 KB
/
Maximize Number of 1's By Flipping M zeroes.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
/*
Input: arr[] = {1, 0, 0, 1, 1, 0, 1, 0, 1, 1, 1}
m = 2
Output: 5 7
We are allowed to flip maximum 2 zeroes. If we flip
arr[5] and arr[7], we get 8 consecutive 1's which is
maximum possible under given constraints
Input: arr[] = {1, 0, 0, 1, 1, 0, 1, 0, 1, 1, 1}
m = 1
Output: 7
We are allowed to flip maximum 1 zero. If we flip
arr[7], we get 5 consecutive 1's which is maximum
possible under given constraints.
Input: arr[] = {0, 0, 0, 1}
m = 4
Output: 0 1 2
Since m is more than number of zeroes, we can flip
all zeroes.
*/
#include<iostream>
using namespace std;
int main() {
//code
int t,n,m,m1,i;
cin>>t; // No of test cases
while(t--)
{
cin>>n; // no of elements(Only 1 and 0) in array)
int a[n];
for(i=0;i<n;i++)
{
cin>>a[i]; // scan the elements
}
cin>>m; // no of zeroes allowed to be flipped
m1=m;
for(i=0;i<n&&m1;i++) // finding index of first 1 after m zeroes
{
if(a[i] == 0)
{
m1--;
}
}
if(i==n) // if zeroes present in array are less than m then whole array length is answer
{
cout<<n<<endl;
continue;
}
int start = 0;
int curr = i;
int max = -1;
while(curr<n) // here sliding window techique is used,curr will start from first 1 after first m zeroes
{
if(a[curr] == 0) // if zero is found that means we have found m+1 zeroes from start,so update max now
{
max = max > (curr-start) ? max : (curr-start);
for(;a[start]!=0;start++){} // move the start from first 0 of m zeroes to next element
start++;
}
else if(curr == n-1) // if we are at last element and after m zeroes from start (m+1)th zero is not found then update max
{
max = max > (curr-start+1) ? max : (curr-start+1);
}
curr++;
}
cout<<max<<endl;
// max = max > curr-start ? max : curr-start;
}
return 0;
}