说在前面
再乱用指针me吔屎
题目
题目大意
给出一个字符串
a
a
,再给出一个长度 。对于
i∈[ 0,|a| ]
i
∈
[
0
,
|
a
|
]
的每个
i
i
,求出长度为 且与
a
a
的最长公共子序列的长度为 的字符串有多少个,答案对
109+7
10
9
+
7
取模
范围:
|a|≤15,n≤1000
|
a
|
≤
15
,
n
≤
1000
,字符集
={A,G,T,C}
=
{
A
,
G
,
T
,
C
}
输入输出格式
输入格式:
第一行一个整数
T
T
,表示数据组数,保证
对于每组数据,包含两行:
第一行一个字符串表示
a
a
第二行一个整数表示
输出格式:
对于每组数据,输出
|a|+1
|
a
|
+
1
行:
第
i
i
行表示与 的LCS的长度为
i−1
i
−
1
的串总数
解法
感觉是一个比较好玩的dp套dp
首先看到这个题,就应该能反应过来是dp套dp,因为外层的统计显然不可能是字符串算法,内层求LCS只有dp做法
那么考虑一下怎么套。我们发现,在LCS的dp矩阵中,相邻的格子最多相差
1
1
,所以我们可以差分之后状压。
于是我们可以预处理出 每个状态在某个字符的转移,然后定义表示已经选了
i
i
个字符,当前差分状态为 的方案数,转移:
∀j:dp[i][trans[se][j]]+=dp[i−1][se]
∀
j
:
d
p
[
i
]
[
t
r
a
n
s
[
s
e
]
[
j
]
]
+
=
d
p
[
i
−
1
]
[
s
e
]
,需要开滚动
然后这题就做完了
然后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() ;
}
}