2019牛客暑期多校训练营(第三场)B题、H题
题意:
就是说给你一个由0或1组成的字符串,让你找出来一个0的数量和1的数量相等的最长子字符串和最长子序列
题解:
可以把0当作-1,把1当作1来计算字符串的前缀和
这样的话,当两个位置的前缀和的值相同的时候,那么这两个位置中间的部分就满足题意,除此之外前缀和为0的地方也满足题意
因为这是前缀和,当两个位置前缀和相同就可以让后一个减去前一个前缀和,那么这中间一段便是0和1相同的长度
为了防止出现负数下标,可以让计算出来的前缀和加上一个大正数
代码一:
1 #include<stdio.h> 2 #include<string.h> 3 #include<stdlib.h> 4 #include<algorithm> 5 using namespace std; 6 char a[100005]; 7 int ans[200005][2]; 8 int sum[100005]; 9 int main() 10 { 11 int n; 12 int s1=0,s0=0,maxx=0; 13 sum[0]=0; 14 scanf("%d",&n); 15 scanf("%s",a); 16 ans[100000][0]=-1; 17 for(int i=0;i<n;i++) 18 { 19 if(a[i]=='1') 20 { 21 s1++; 22 sum[i]=sum[i-1]+1; 23 } 24 else 25 { 26 s0++; 27 sum[i]=sum[i-1]-1; 28 } 29 if(ans[sum[i]+100000][0]==0) 30 { 31 ans[sum[i]+100000][0]=i; 32 } 33 else 34 { 35 ans[sum[i]+100000][1]=i; 36 37 maxx=max(maxx,ans[sum[i]+100000][1]-ans[sum[i]+100000][0]); 38 } 39 } 40 if(s0==s1) 41 { 42 printf("%d %d ",s0+s1,s0+s1); 43 } 44 else if(s0<s1) 45 { 46 printf("%d %d ",maxx,s0*2); 47 } 48 else if(s0>s1) 49 { 50 printf("%d %d ",maxx,s1*2); 51 } 52 return 0; 53 }
代码二:
1 #include <bits/stdc++.h> 2 3 #define ll long long 4 5 using namespace std; 6 7 const int N=1e5+10; 8 9 int pre[2*N]; 10 11 char s[N]; 12 13 int main(void) 14 15 { 16 17 int n,sum=0,cnt0=0,cnt1=0,ans=0; 18 19 scanf("%d%s",&n,s+1); 20 21 for(int i=1;i<=n;i++) 22 23 { 24 25 if(s[i]=='0') cnt0++,sum++; 26 27 else cnt1++,sum--; 28 29 if(sum==0) 30 31 { 32 33 if(ans<i) ans=i; 34 35 continue; 36 37 } 38 39 if(pre[sum+n]==0) pre[sum+n]=i; 40 41 else if(i-pre[sum+n]>ans) 42 43 ans=i-pre[sum+n]; 44 45 } 46 47 printf("%d %d ",ans,min(cnt0,cnt1)*2); 48 49 return 0; 50 51 }