01串(51Nod-1391)
生活随笔
收集整理的這篇文章主要介紹了
01串(51Nod-1391)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目
給定一個01串S,求出它的一個盡可能長的子串S[i..j],滿足存在一個位置i<=x <j, S[i..x]中0比1多,而S[x + 1..j]中1比0多。求滿足條件的最長子串長度。
輸入
一行包含一個只由0和1構成的字符串S。 S的長度不超過1000000。
輸出
一行包含一個整數,表示滿足要求的最長子串的長度。
輸入樣例
10
輸出樣例
0
思路:
通過題目可以看出,起碼需要尋找兩次,一次從前向后找某個位置 0 個個數大于 1 的個數的最大長度,一次從后往前找某個位置 1 的個數大于 0 的個數
由于題目中給的是 01 串,尋找滿足條件的最長子串長度并不方便,因此可將 0 轉為 -1,然后再去求序列的前綴和,這樣一來,一段序列的和小于 0,可以說明 0 的個數大于 1,一段序列的和大于 0,可以說明 1 的個數大于 0,于是問題就轉化為求到某個位置大于 0 或小于 0 的最長長度
源程序
#include<iostream> #include<cstdio> #include<cstdlib> #include<string> #include<cstring> #include<cmath> #include<ctime> #include<algorithm> #include<utility> #include<stack> #include<queue> #include<vector> #include<set> #include<map> #define EPS 1e-9 #define PI acos(-1.0) #define INF 0x3f3f3f3f #define LL long long const int MOD = 1E9+7; const int N = 1000000+5; const int dx[] = {0,0,-1,1,-1,-1,1,1}; const int dy[] = {-1,1,0,0,-1,1,-1,1}; using namespace std;int a[N],pos[N]; int suf[N],sum[N]; int leftPos[N],rightPos[N]; int main() {string str;cin>>str;int n=str.size();for(int i=1; i<=n; i++) {if(str[i-1]=='0')//0轉為-1a[i]=-1;elsea[i]=1;sum[i]=sum[i-1]+a[i];//計算前綴和}memset(pos,-1,sizeof(pos));for(int i=1; i<=n; i++) {//從前向后找if(sum[i]<0)//記錄前綴小于0的位置leftPos[i]=i;else {if(pos[sum[i]+1]!=-1)leftPos[i]=i-pos[sum[i]+1];else {leftPos[i]=0;pos[sum[i]]=i;}}}memset(pos,-1,sizeof(pos));for(int i=n; i>=1; i--) {//從后向前找suf[i]=suf[i+1]+a[i];//計算后綴和if(suf[i]>0)//記錄后綴大于0的位置rightPos[i]=n-i+1;else {if(pos[-(suf[i]-1)]!=-1)rightPos[i]=pos[-(suf[i]-1)]-i;else {rightPos[i]=0;pos[-(suf[i])]=i;}}}int res=0;for(int i=1; i<n; i++)if(leftPos[i]>0&&rightPos[i+1]>0)//枚舉前綴大于0后綴大于0的個數res=max(rightPos[i+1]+leftPos[i],res);//記錄長度cout<<max(0,res)<<endl;return 0; }?
總結
以上是生活随笔為你收集整理的01串(51Nod-1391)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 树形结构 —— 树与二叉树 —— 树的重
- 下一篇: 动态规划 —— 线性 DP —— 序列问