HDU6170【DP+树状数组+差分维护】

哦，这题我1A诶，好开心啊，写个题解嘛

先预处理一下第二个串，预处理出每个位置字母相同的向右最远位置。然后大力DP!
分成s串(1串)，p串(2串)，dp[i][j]代表从p串位置1-i能否匹配s串1-j，然后需要维护就是对于”*”，因为存在p串的第 i 个字符作用还可以匹配位置 j 后面的元素，不能再枚举一层，考虑前缀和，然后用树状数组差分即可~

具体细节见代码注释~Thanks for watching!

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
#define mem(a, b) memset(a, b, sizeof(a))

const int N = 2550;
char s[N], p[N], pp[N];
int f[N], num, n, pos[N];
int dp[N][N];

//预处理 
void init(){
    mem(f, 0);
    int len = strlen(pp);
    num = 0;
    for(int i=0;i<len;i++){
        if(pp[i] == '*')    //对*位置的字符标记 
            f[num] = 1;
        else p[++num] = pp[i];
    }
    n = strlen(s+1);
    pos[n] = n;         //pos[i]是i位置相同字符向后最远的位置 
    for(int i=n-1;i>=1;i--){
        if(s[i] == s[i+1]) pos[i] = pos[i+1];
        else pos[i] = i;
    }
}

//树状数组 
//细节：考虑树状数组0位置无意义，但是我DP初始化需要dp[0][0]=1，所以所有坐标+1. 
inline int lowbit(int x){return x & -x;}
int sum(int i, int j){
    i++;
    j++;
    int ans = 0;
    while(j > 0){
        ans = ans + dp[i][j];
        j = j - lowbit(j);
    }
    return ans;
}
void add(int i, int j, int val){
    i++;
    j++;
    while(j <= n+1){
        dp[i][j] = dp[i][j] + val;
        j = j + lowbit(j);
    }
}


void solve(){
    //初始化 
    for(int i=0;i<=num+1;i++)
    for(int j=0;j<=n+1;j++)
        dp[i][j] = 0;
    add(0, 0, 1);
    add(0, 1, -1);

    for(int i=1;i<=num;i++){
        //如果是 .  
        if(p[i] == '.'){
            if(f[i]){       //且如果是可以任意长度 
                for(int j=n;j>=0;j--){
                    if(sum(i-1, j)){        //先去处理空的 
                        add(i, j, 1);
                        add(i, j+1, -1);
                    }
                    if(sum(i-1, j-1) && j){     //然后处理[j, pos[j]] 
                        add(i, j, 1);
                        add(i, pos[j]+1, -1);
                    }
                }
            }
            else{
                //这里只有一个字符 （可任意 
                for(int j=n;j>=1;j--)
                    if(sum(i-1, j-1)){
                        add(i, j, 1);
                        add(i, j+1, -1);
                    }
            }
        }
        else if(f[i]){
            //处理非 . 字符的可任意长度 
            for(int j=n;j>=0;j--){
                if(sum(i-1, j)){    //先处理空的 
                    add(i, j, 1);
                    add(i, j+1, -1);
                }
                if(sum(i-1, j-1) && j){     //然后处理[j, pos[j]] 
                    if(p[i] == s[j]){
                        add(i, j, 1);
                        add(i, pos[j]+1, -1);
                    }
                }
            }
        }
        else{
            //这里只有一个字符 
            for(int j = n;j>=1;j--)
                if(sum(i-1, j-1) && p[i] == s[j]){
                        add(i, j, 1);
                        add(i, j+1, -1);
                }
        }
    }
    if(sum(num, n)) puts("yes");
    else puts("no");
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%s%s", s+1, pp);
        init();
        solve();
    }
    return 0;
}