1. 程式人生 > >Two strings hdu6170 dp遞推

Two strings hdu6170 dp遞推

題意:給出兩個字串,第一個字串只包含小寫或者大寫字母,第二個字串包含小寫或者大寫字母或者特殊字元“.”和“*”,這裡“.”可以替換為任意字元,但是不能變成空。
這裡“a*”可以變成空串,可以變成a,也可以是aa,也可以是aaa,還可以是aaaa.以此類推,不限長度。

問第二個串和第一個串是否能夠完全相等。

思路:dp記錄狀態,寫記憶化超時了,所以只能正著推了,設d[i][j][k]為第一個串0-i和第二個串0-j狀態為k時是否能匹配,k=1時代表可以使用若干個s2[j],也就是說,只在當前s2[j]為‘*’或者後一位為‘*’時狀態更新。不太好說,意思表達不出來,詳見程式碼。

#include <iostream>  
#include <cstdio>   
#include <cstring>  
#include <algorithm>  
#include <vector>  
#include <map>  
#include <cmath>  
#include <set>  
#include <queue>  
using namespace std;  
typedef long long ll;    
const int INF=1e9+100;     
const int mod=1e9+7;

const int N=2502;
bool d[N][N][2];
char s1[N],s2[N];
int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%s%s",s1+1,s2+1);
        int len1=strlen(s1+1),len2=strlen(s2+1);
        d[0][0][1]=d[0][0][0]=1;
        for(int i=1;i<=len2;i++) d[0][i][1]=d[0][i][0]=1;
        for(int i=1;i<=len1;i++){
            for(int j=1;j<=len2;j++){
                d[i][j][1]=d[i][j][0]=0;
                if(s2[j]=='*'){
                    d[i][j][0]=d[i][j][0]|d[i][j-2][0]|d[i][j-2][1];
                    d[i][j][1]=d[i][j][1]|d[i][j-1][1]|d[i][j-1][0];
                }else if(s2[j]=='.'){
                    if(s2[j+1]=='*'){
                        if(d[i-1][j][1]==1&&s1[i]==s1[i-1])
                            d[i][j][1]=1;
                        d[i][j][1]=d[i][j][1]|d[i-1][j-1][0]|d[i-1][j-1][1];
                        d[i][j][0]=d[i][j][0]|d[i][j-1][0]|d[i][j-1][1];
                    }else{
                        d[i][j][0]=d[i][j][0]|d[i-1][j-1][0]|d[i-1][j-1][1];
                    }
                }else{
                    if(s2[j+1]=='*'){
                        if(d[i-1][j][1]==1&&s1[i]==s2[j])
                            d[i][j][1]=1;
                        d[i][j][1]=d[i][j][1]|d[i-1][j-1][0]|d[i-1][j-1][1];
                        if(s1[i]!=s2[j])
                            d[i][j][1]=0;
                        d[i][j][0]=d[i][j][0]|d[i][j-1][0]|d[i][j-1][1];
                    }else{
                        d[i][j][0]=d[i][j][0]|d[i-1][j-1][0]|d[i-1][j-1][1];
                        if(s1[i]!=s2[j])
                            d[i][j][0]=0;
                    }
                } 
            }
        }
        if(d[len1][len2][1]||d[len1][len2][0])
            printf("yes\n");
        else
            printf("no\n");
    }
    return 0;
}