[BZOJ3864]-Hero meet devil-dp套dp

说在前面

再乱用指针me吔屎

---

题目

BZOJ3864传送门

题目大意

给出一个字符串 $a$，再给出一个长度 $n$。对于$i\in[~0,|a|~]$的每个 $i$，求出长度为 $n$ 且与 $a$ 的最长公共子序列的长度为 $i$ 的字符串有多少个，答案对 $10^9+7$ 取模
范围：$|a|\leq 15,n\leq 1000$，字符集$=\{A,G,T,C\}$

输入输出格式

输入格式：
第一行一个整数 $T$ ，表示数据组数，保证$T\leq 5$
对于每组数据，包含两行：
第一行一个字符串表示 $a$
第二行一个整数表示 $n$

输出格式：
对于每组数据，输出$|a|+1$行：
第 $i$ 行表示与 $a$ 的LCS的长度为 $i-1$ 的串总数

---

解法

感觉是一个比较好玩的dp套dp
首先看到这个题，就应该能反应过来是dp套dp，因为外层的统计显然不可能是字符串算法，内层求LCS只有dp做法

那么考虑一下怎么套。我们发现，在LCS的dp矩阵中，相邻的格子最多相差 $1$，所以我们可以差分之后状压。
于是我们可以预处理出 每个状态在某个字符的转移，然后定义$dp[i][se]$表示已经选了 $i$ 个字符，当前差分状态为 $se$ 的方案数，转移：$\forall j:dp[i][trans[se][j]]+=dp[i-1][se]$，需要开滚动

然后这题就做完了

然后me样例都过不了，因为me一开始写成了这样：

int *now = dp[0] , *pre = dp[1] ;
memset( now , 0 , sizeof( now ) ) ;

而且me几乎一直都是这么用的…直到今天，me才发现sizeof(now)是4 = =？？？
正确写法是这样的：

int (*now)[32768] = &dp[0] , (*pre)[32768] = &dp[1] ;
memset( now , 0 , sizeof( *now ) ) ;

---

下面是代码

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std ;

const int P = 1e9 + 7 ;
int T , N , lens , popcnt[1<<15] ;
int trans[1<<15][4] , dp[2][1<<15] , id[127] ;
char s[20] ;

void sadd( int &a , int b ){
    a += b ;
    if( a >= P ) a -= P ;
}

void preWork(){
    int tmp[20] , d[20] ; tmp[0] = d[0] = 0 ;

    for( int se = 0 , lim = 1 << lens ; se < lim ; se ++ ){
        if( se ) popcnt[se] = popcnt[ se^(se&-se) ] + 1 ;
        for( int i = 1 ; i <= lens ; i ++ )
            tmp[i] = tmp[i-1] + (bool)( se & ( 1 << ( i - 1 ) ) ) ;
        for( int i = 0 ; i < 4 ; i ++ ){
            trans[se][i] = 0 ;
            memset( d , 0 , sizeof( d ) ) ;
            for( int j = 1 ; j <= lens ; j ++ ){
                d[j] = max( tmp[j] , d[j-1] ) ;
                if( i == s[j] ) d[j] = max( d[j] , tmp[j-1] + 1 ) ;
            }
            for( int j = 0 ; j < lens ; j ++ )
                if( d[j] != d[j+1] ) trans[se][i] |= ( 1 << j ) ;
        }
    }
}

int ans[20] ;
void solve(){
    int (*now)[32768] = &dp[0] , (*pre)[32768] = &dp[1] ;
    memset( now , 0 , sizeof( *now ) ) , now[0][0] = 1 ;
    for( int i = 1 ; i <= N ; i ++ ){
        swap( now , pre ) ;
        memset( now , 0 , sizeof( *now ) ) ;
        for( int se = 0 , lim = 1 << lens ; se < lim ; se ++ )
            for( int j = 0 ; j < 4 ; j ++ )
                sadd( now[0][ trans[se][j] ] , pre[0][se] ) ;
    } memset( ans , 0 , sizeof( ans ) ) ;
    for( int se = 0 , lim = 1 << lens ; se < lim ; se ++ )
        sadd( ans[ popcnt[se] ] , now[0][se] ) ;
    for( int i = 0 ; i <= lens ; i ++ )
        printf( "%d\n" , ans[i] ) ;
}

int main(){
    id['A'] = 0 , id['G'] = 1 , id['T'] = 2 , id['C'] = 3 ;
    scanf( "%d" , &T ) ;
    while( T -- ){
        scanf( "%s%d" , s + 1 , &N ) , lens = strlen( s + 1 ) ;
        for( int i = 1 ; i <= lens ; i ++ ) s[i] = id[ s[i] ] ;
        preWork() ; solve() ;
    }
}