题目
有一个字符串,你可以修改字符串中的某一个字符,要求使得修改后"udu"的子序列个数最少。
思路
要完成题目的任务,我们需要找到众多udu子序列中重合最多的d或u是哪一个,把它改掉就好了。因此我们分三种情况讨论/
第一种情况是我们修改的字符是'd',这时候包含这个字符'd'的udu子序列总共有:它前面的'u'的数量乘它后面的'u'的数量,我们枚举每一个'd'计算,就能找出修改'd'最多能减少多少个'udu'子序列。
第二种情况是我们修改的字符是做子序列开头的'u',包含它的udu子序列无非就是后面的每个'd'的后面有多少个'u'之和,同样枚举所有u求最大。
第三种情况就是修改的字符是子序列结尾的'u',包含它的udu子序列就是前面的每个'd'的前面有多少个'u'之和,同样枚举求最大。
代码实现上,只需要一个front数组和一个later数组,front[i]表示str[i]上如果是'd',其前面有多少个'u',later[i]表示str[i]上如果是'd',其后面有多少个'u',当然如果不是'd',就赋成0,因此第一种情况就解决了,第二种情况也就是later数组的后缀和,第三种情况也就是front数组的前缀和,问题也就解决了。
代码
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxn=2e5+5;
int front[maxn],later[maxn];
signed main(){
string s;cin>>s;
int cnt=0;
int ans=0,idx=-1;
for(int i=0;i<s.size();i++){
if(s[i]=='u')cnt++;
else if(s[i]=='d')front[i]=cnt;
}
cnt=0;
for(int i=s.size()-1;i>=0;i--){
if(s[i]=='u')cnt++;
else if(s[i]=='d')later[i]=cnt;
}
for(int i=0;i<s.size();i++){
if(front[i]*later[i]>ans){
ans=front[i]*later[i];
idx=i;
}
}
for(int i=1;i<s.size();i++){
front[i]+=front[i-1];
}
for(int i=s.size()-2;i>=0;i--){
later[i]+=later[i+1];
}
for(int i=0;i<s.size();i++){
if(s[i]=='u'){
if(ans<later[i]){
ans=later[i];
idx=i;
}
break;
}
}
for(int i=s.size()-1;i>=0;i--){
if(s[i]=='u'){
if(ans<front[i]){
ans=front[i];
idx=i;
}
break;
}
}
if(idx!=-1)s[idx]='v';
cout<<s;
}