1. 程式人生 > >POJ 2192 Zipper (dp)

POJ 2192 Zipper (dp)

 連結: http://poj.org/problem?id=2192 

題意:就是給定三個字串A,B,C;判斷C能否由AB中的字元組成,同時這個組合後的字元順序必須是A,B中原來的順序,不能逆序;例如:A:mnl,B:xyz;如果C為mnxylz,就符合題意;如果C為mxnzly,就不符合題意,原因是z與y順序不是B中順序。


DP求解:定義dp[i][j]表示A中前i個字元與B中前j個字元是否能組成C中的前 (i+j) 個字元,如果能,標記1,如果不能,標記0;

有了這個定義,我們就可以找出狀態轉移方程了,初始狀態dp[0][0] = 1:

如果 dp[i-1][j] == 1 && C[i+j-1] == A[i-1]       ---->    

dp[i][j] = 1 
如果 dp[i][j-1] == 1 && C[i+j-1] == B[j-1]       ---->     
dp[i][j] = 1 

程式碼如下:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <stack>
#include <set>
#include <cmath>
#include <queue>
#define MAXN 1010
#define Mul(x)((x)*(x))
#define RST(N)memset(N, 0, sizeof(N))
using namespace std;

int dp[MAXN][MAXN], cas;
char A[MAXN], B[MAXN], C[MAXN];

int main()
{
    int kcas = 0;
    scanf("%d", &cas);
    while(cas--) {
        scanf("%s %s %s", A, B, C);
        RST(dp);
        int La = strlen(A), Lb = strlen(B);
        dp[0][0] = 1;
        for(int i=0; i<=La; i++) {
            for(int j=0; j<=Lb; j++) {
                if(i > 0 && dp[i-1][j] == 1 && C[i+j-1] == A[i-1]) {
                    dp[i][j] = 1;
                }
                if(j > 0 && dp[i][j-1] == 1 && C[i+j-1] == B[j-1]) {
                    dp[i][j] = 1;
                }
            }
        }
        printf("Data set %d: %s\n", ++kcas, dp[La][Lb] ? "yes" : "no");
    }
    return 0;
}