CodeForces 1108 D Diverse Garland dp 输出路径 滚动数组

TP

题意: 给出一个只包含'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;
}