1124 - 咸鱼魔法记
DESCRIPTION
给你一个01串,我们定义这个串的咸鱼值,是最长的全1串。现在你最多可以使用K次咸鱼魔法,每次魔法,你可以使得一个位置翻转(0变成1,1变成0)。问你这个串的咸鱼值最多是多少。
INPUT
第一行两个整数N,K。表示串的长度和可以施展咸鱼魔法的次数。(N,K<=300000) 第二行N个01整数。
OUTPUT
输出答案。
SAMPLE INPUT
10 2
1 0 0 1 0 1 0 1 0 1
SAMPLE OUTPUT
5
标记0的坐标,求取最长区间
#include <iostream> #include <algorithm> #include <cstring> #include <cstdio> #include <vector> #include <queue> #include <cmath> #include <ctime> #include <map> #include <set> using namespace std; #define lowbit(x) (x&(-x)) #define max(x,y) (x>y?x:y) #define min(x,y) (x<y?x:y) #define MAX 100000000000000000 #define MOD 1000000007 #define pi acos(-1.0) #define ei exp(1) #define PI 3.141592653589793238462 #define INF 0x3f3f3f3f3f #define mem(a) (memset(a,0,sizeof(a))) typedef long long ll; int n,k,x; int a[300010]; int main() { scanf("%d%d",&n,&k); a[0]=0; int ans=-1,pos=0,cnt=1; for(int i=1;i<=n;i++) { scanf("%d",&x); if(x==0) { a[cnt++]=i; ans=max(ans,pos); pos=0; } else pos++; } if(k==0) printf("%d\n",ans); else if(k>=(cnt-1)) printf("%d\n",n); else { for(int i=k+1;i<cnt;i++) { ans=max(ans,a[i]-a[i-k-1]-1); } printf("%d\n",ans); } return 0; }