题目:http://www.lydsy.com/JudgeOnline/problem.php?id=1566
事实上求sigma(ai^2)可以理解成求操作序列A与操作序列B产生的结果一样的(A,B)的对数,那么DP,令dp[i][][k][h]表示A上面取了i个,下面取了j个,B上面取了k个,下面取了h个的方案数,然后DP,然后h这一维可以压掉。
代码:
#include <cstdio>
#define rep( i , l , r ) for ( int i = l ; i <= r ; ++ i )
#define update( x , y ) ( x += y ) %= mod ;
const int mod = 1024523 , maxn = 510 ;
int dp[ maxn ][ maxn ][ maxn ] , n , m ;
char s[ 2 ][ maxn ] ;
int main( ) {
scanf( "%d%d" , &n , &m ) ;
scanf( "%s%s" , s[ 0 ] + 1 , s[ 1 ] + 1 ) ;
rep( i , 0 , n ) rep( j , 0 , m ) rep( k , 0 , n ) dp[ i ][ j ][ k ] = 0 ;
dp[ 0 ][ 0 ][ 0 ]= 1 ;
rep( i , 0 , n ) rep( j , 0 , m ) rep( k , 0 , n ) if ( dp[ i ][ j ][ k ] ) {
if ( s[ 0 ][ i + 1 ] == s[ 0 ][ k + 1 ] ) update( dp[ i + 1 ][ j ][ k + 1 ] , dp[ i ][ j ][ k ] ) ;
if ( s[ 0 ][ i + 1 ] == s[ 1 ][ i + j - k + 1 ] ) update( dp[ i + 1 ][ j ][ k ] , dp[ i ][ j ][ k ] ) ;
if ( s[ 1 ][ j + 1 ] == s[ 0 ][ k + 1 ] ) update( dp[ i ][ j + 1 ][ k + 1 ] , dp[ i ][ j ][ k ] ) ;
if ( s[ 1 ][ j + 1 ] == s[ 1 ][ i + j - k + 1 ] ) update( dp[ i ][ j + 1 ][ k ] , dp[ i ][ j ][ k ] ) ;
}
printf( "%d\n" , dp[ n ][ m ][ n ] ) ;
return 0 ;
}