Codeforces 803E--Roma and Poker （DP）

原题链接：http://codeforces.com/problemset/problem/803/E

题意：给一个n长度的字符串，其中'?'可以替换成'D'、'W'、'L'中的任意一种，'D'等价于0, 'W'等价于1、'L'等价于-1。输出所有'?'被替换掉后，W和L的数目之差为k，且任意一个[1, i]的子串中W和L数目之差不能等于k。

思路：用DP做。定义bool dp[i][j]代表前i个字符W和L数目之差为j, -k<=j<=k(在数组中范围为[0, 2*k])，那么当str[i]为'D'时dp[i][j]转移到dp[i-1][j], 为'W'时dp[i][j]转移到dp[i-1][j+1], str[i]为'D'时dp[i][j]转移到dp[i-1][j-1], 初始值dp[0][0]为true。

接着用一遍dfs倒推求结果，注意字符加的位置。

AC代码：

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #include<string>
 using namespace std;
 int dp[][];
 int n,k;
 string str;
 void change(int i, int L, int D, int W){
         for(int j=;j<*k;j++){
             if(dp[i-][j]){
                 if(D) dp[i][j]=;
                 if(L){
                     if(i!=n&&j-==)//[1, i]子串中W和L数目之差不能等于k
                         continue;
                     dp[i][j-]=;
                 }
                 if(W){
                     if(i!=n&&j+==*k)
                         continue;
                     dp[i][j+]=;
                 }
         }
     }
     return;
 }
 string ss;
 //int t=0;
 bool res(int i, int j, string ans){
     //t++;
     if(i==&&j==k){
         ss=ans;
         return ;
     }
     if(str[i-]!='?'){
         if(str[i-]=='D') return res(i-, j, 'D'+ans);
         if(str[i-]=='W') return res(i-, j-, 'W'+ans);
         if(str[i-]=='L') return res(i-, j+, 'L'+ans);
     }
     else
     {
         if(dp[i-][j]&&res(i-, j, 'D'+ans)) return ;
         if(dp[i-][j-]&&res(i-, j-, 'W'+ans)) return ;
         if(dp[i-][j+]&&res(i-, j+, 'L'+ans)) return ;
     }
 
     return ;
 }
 int main()
 {
     while(cin>>n>>k)
     {
         memset(dp, , sizeof(dp));
         dp[][k]=;
         cin>>str;
         if(str[n-]=='D'){
             cout<<"NO"<<endl;
             continue;
         }
         for(int i=;i<=n;i++){
             if(str[i-]=='?')
                 change(i, , , );
             else if(str[i-]=='D')
                 change(i, , , );
             else if(str[i-]=='W')
                 change(i, , , );
             else
                 change(i, , , );
         }
         string ans;
         if(dp[n][]){
             res(n, , ans);
             cout<<ss<<endl;
         }
         else if(dp[n][*k]){
             res(n, *k, ans);
             cout<<ss<<endl;
         }
         else
             cout<<"NO"<<endl;
         //cout<<t<<endl;
     }
     return ;
 }

这代码调了我好久啊QAQ，感觉自己真菜