1. 程式人生 > >Codeforces 1082B Vova and Trophies(字首+字尾)

Codeforces 1082B Vova and Trophies(字首+字尾)

題目連結:Vova and Trophies

題意:給定長度為n的字串s,字串中只有G和S,只允許最多一次操作:任意位置的兩個字元互換。求連續G的最長長度。

題解:維護pre和pr,nxt和nx。pre和nxt像連續子段和一樣維護G的個數,pr和nx維護分別前面和後面各有多少個G。

遍歷每個位置i,pre[i-1]+nxt[i+1],判斷當前位置i能不能用多餘的G來替換,pr和nx判斷一下是否與pre和nxt相同即可,過程中更新答案。

 1 #include <cstdio>
 2 #include <algorithm>
 3 using
namespace std; 4 5 const int N=2e5+10; 6 typedef long long ll; 7 8 char s[N]; 9 ll pr[N],nx[N]; 10 ll pre[N],nxt[N]; 11 12 int main(){ 13 int n; 14 ll ans=0; 15 scanf("%d%s",&n,s+1); 16 for(int i=1;i<=n;i++){ 17 pr[i]=pr[i-1]; 18 if(s[i]=='
G') pre[i]=pre[i-1]+1,pr[i]++; 19 else pre[i]=0; 20 } 21 for(int i=n;i>=1;i--){ 22 nx[i]=nx[i+1]; 23 if(s[i]=='G') nxt[i]=nxt[i+1]+1,nx[i]++; 24 else nxt[i]=0; 25 } 26 for(int i=1;i<=n;i++){ 27 ll sum=0; 28 sum=pre[i-1]+nxt[i+1
]; 29 if(s[i]=='G') ans=max(ans,sum+1); 30 else{ 31 if(pr[i-1]!=pre[i-1]||nx[i+1]!=nxt[i+1]) sum++; 32 ans=max(ans,sum); 33 } 34 } 35 printf("%lld\n",ans); 36 return 0; 37 }
View Code