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 
#include
#include
#include
#include
#include
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;idp[n][i]){
            ans=dp[n][i];
            ansi=i;
        }
    }

    printf("%d\n",ans);

    stacks;
    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
#define fuck(x) std::cout<<"["<<#x<<"->"<dp[n%2][i]){
            ans=dp[n%2][i];
            ansi=i;
        }
    }

    printf("%d\n",ans);
    
    //路径输出
    stacks;
    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;
}

 

你可能感兴趣的:(dp)