题意: 给出一个只包含'R' 'G' 'B' 的序列, 问最少改变多少个字符能把这个序列改成没有相同字符相邻的形式
思路: dp dp[i][j]表示走到第i个位置时(前i位已符合条件), 结尾是j(0:R, 1:G, 2:B)时, 最少要改变多少个字符.
递推式:
当a[i]=0时
dp[i][0]=min(dp[i-1][1],dp[i-1][2])
dp[i][1]=min(dp[i-1][0],dp[i-1][2])+1
dp[i][2]=min(dp[i-1][0],dp[i-1][1])+1
a[i]=1或2时与之类似.
同时因为i状态只与i-1状态有关, 可以用滚动数组优化空间, 虽然也无所谓...
代码:
#include <iostream>
#include<vector>
#include<cstdio>
#include<map>
#include<algorithm>
#include<stack>
using namespace std;
const int M=2e5+5;
const int inf=1e9+5;
char x[M];
int a[M];
int dp[M][3];
int bf[M][3];
int main()
{
int n;
scanf("%d",&n);
scanf("%s",x);
for(int i=0;i<n;i++){
if(x[i]=='R')a[i+1]=0;
if(x[i]=='G')a[i+1]=1;
if(x[i]=='B')a[i+1]=2;
}
for(int i=1;i<=n;i++){
if(a[i]==0){
dp[i][0]=min(dp[i-1][1],dp[i-1][2]);
if(dp[i][0]==dp[i-1][1])bf[i][0]=1;
else bf[i][0]=2;
dp[i][1]=min(dp[i-1][0],dp[i-1][2])+1;
if(dp[i][1]==dp[i-1][0]+1)bf[i][1]=0;
else bf[i][1]=2;
dp[i][2]=min(dp[i-1][0],dp[i-1][1])+1;
if(dp[i][2]==dp[i-1][0]+1)bf[i][2]=0;
else bf[i][2]=1;
}
if(a[i]==1){
dp[i][1]=min(dp[i-1][0],dp[i-1][2]);
if(dp[i][1]==dp[i-1][0])bf[i][1]=0;
else bf[i][1]=2;
dp[i][0]=min(dp[i-1][1],dp[i-1][2])+1;
if(dp[i][0]==dp[i-1][1]+1)bf[i][0]=1;
else bf[i][0]=2;
dp[i][2]=min(dp[i-1][0],dp[i-1][1])+1;
if(dp[i][2]==dp[i-1][0]+1)bf[i][2]=0;
else bf[i][2]=1;
}
if(a[i]==2){
dp[i][2]=min(dp[i-1][0],dp[i-1][1]);
if(dp[i][2]==dp[i-1][0])bf[i][2]=0;
else bf[i][2]=1;
dp[i][1]=min(dp[i-1][0],dp[i-1][2])+1;
if(dp[i][1]==dp[i-1][0]+1)bf[i][1]=0;
else bf[i][1]=2;
dp[i][0]=min(dp[i-1][1],dp[i-1][2])+1;
if(dp[i][0]==dp[i-1][1]+1)bf[i][0]=1;
else bf[i][0]=2;
}
}
// int ans=min(min(dp[n][0],dp[n][1]),dp[n][2]);
int ans=inf,ansi;
for(int i=0;i<3;i++){
if(ans>dp[n][i]){
ans=dp[n][i];
ansi=i;
}
}
printf("%d\n",ans);
stack<int>s;
int pos=n;
while(pos>0){
s.push(ansi);
ansi=bf[pos][ansi];
pos--;
}
while(!s.empty()){
int now=s.top();
s.pop();
if(now==0)printf("R");
if(now==1)printf("G");
if(now==2)printf("B");
}
return 0;
}
滚动数组:
#include<bits/stdc++.h>
#define fuck(x) std::cout<<"["<<#x<<"->"<<x<<"]"<<endl;
using namespace std;
typedef long long ll;
const int M=2e5+5;
const int inf=1e9+5;
char a[M];
int dp[2][3];//滚动数组
int bf[M][3];
int main()
{
int n;
scanf("%d",&n);
scanf("%s",a+1);
for(int i=1;i<=n;i++){
if(a[i]=='R')a[i]=0;
if(a[i]=='G')a[i]=1;
if(a[i]=='B')a[i]=2;
}
for(int j=1;j<=n;j++){
int i=j%2;
if(a[j]==0){
dp[i][0]=min(dp[i^1][1],dp[i^1][2]);
if(dp[i][0]==dp[i^1][1])bf[j][0]=1;
else bf[j][0]=2;
dp[i][1]=min(dp[i^1][0],dp[i^1][2])+1;
if(dp[i][1]==dp[i^1][0]+1)bf[j][1]=0;
else bf[j][1]=2;
dp[i][2]=min(dp[i^1][0],dp[i^1][1])+1;
if(dp[i][2]==dp[i^1][0]+1)bf[j][2]=0;
else bf[j][2]=1;
}
if(a[j]==1){
dp[i][1]=min(dp[i^1][0],dp[i^1][2]);
if(dp[i][1]==dp[i^1][0])bf[j][1]=0;
else bf[j][1]=2;
dp[i][0]=min(dp[i^1][1],dp[i^1][2])+1;
if(dp[i][0]==dp[i^1][1]+1)bf[j][0]=1;
else bf[j][0]=2;
dp[i][2]=min(dp[i^1][0],dp[i^1][1])+1;
if(dp[i][2]==dp[i^1][0]+1)bf[j][2]=0;
else bf[j][2]=1;
}
if(a[j]==2){
dp[i][2]=min(dp[i^1][0],dp[i^1][1]);
if(dp[i][2]==dp[i^1][0])bf[j][2]=0;
else bf[j][2]=1;
dp[i][1]=min(dp[i^1][0],dp[i^1][2])+1;
if(dp[i][1]==dp[i^1][0]+1)bf[j][1]=0;
else bf[j][1]=2;
dp[i][0]=min(dp[i^1][1],dp[i^1][2])+1;
if(dp[i][0]==dp[i^1][1]+1)bf[j][0]=1;
else bf[j][0]=2;
}
}
int ans=inf,ansi;
for(int i=0;i<3;i++){
if(ans>dp[n%2][i]){
ans=dp[n%2][i];
ansi=i;
}
}
printf("%d\n",ans);
//路径输出
stack<int>s;
int pos=n;
while(pos>0){
s.push(ansi);
ansi=bf[pos][ansi];
pos--;
}
while(!s.empty()){
int now=s.top();
s.pop();
if(now==0)printf("R");
if(now==1)printf("G");
if(now==2)printf("B");
}
return 0;
}