1566: [NOI2009]管道取珠 - BZOJ

Description

Input
第一行包含两个整数n, m，分别表示上下两个管道中球的数目。 第二行为一个AB字符串，长度为n，表示上管道中从左到右球的类型。其中A表示浅色球，B表示深色球。 第三行为一个AB字符串，长度为m，表示下管道中的情形。
Output
仅包含一行，即为 Sigma(Ai^2) i从1到k 除以1024523的余数。
Sample Input
2 1
AB
B

Sample Output
5
HINT

样例即为文中(图3)。共有两种不同的输出序列形式，序列BAB有1种产生方式，而序列BBA有2种产生方式，因此答案为5。
【大致数据规模】
约30%的数据满足 n, m ≤ 12；
约100%的数据满足n, m ≤ 500。

 

 

这个思路太好了，神了

我们要求Ai的平方，就是要求效果相同的取出序列两两组合有多少种方案

相当于有两个人在做这个游戏，你要求他们两个输出序列相同的方案数

这样问题就转化成了求相同的取出序列有多少个，设f[x,y,x',y']表示第一行取到x个，第二行取到y个和第一行取到x'个，第二行取到y'个相同的序列有多少个

最后答案就是f[n,m,n,m]，因为x+y=x'+y'所以我们可以减少一维变成f[x,y,x']还可以用滚动数组优化空间（貌似bzoj上给的空间够大，直接开不会爆）

const
    maxn=550;
    h=1024523;
var
    f:array[0..1,-1..maxn,-1..maxn]of longint;
    a,b:array[0..maxn]of char;
    n,m,xi,yi:longint;
 
function min(x,y:longint):longint;
begin
    if x<y then exit(x);
    exit(y);
end;
 
procedure main;
var
    i,j,l:longint;
begin
    readln(n,m);
    for i:=1 to n do read(a[i]);readln;
    for i:=1 to m do read(b[i]);
    f[0,0,0]:=1;xi:=0;yi:=1;
    for l:=1 to n+m do
        begin
            for i:=0 to min(n,l) do
                for j:=0 to min(n,l) do
                    begin
                        if a[i]=a[j] then inc(f[yi,i,j],f[xi,i-1,j-1]);
                        if b[l-i]=b[l-j] then inc(f[yi,i,j],f[xi,i,j]);
                        if a[i]=b[l-j] then inc(f[yi,i,j],f[xi,i-1,j]);
                        if b[l-i]=a[j] then inc(f[yi,i,j],f[xi,i,j-1]);
                        f[yi,i,j]:=f[yi,i,j]mod h;
                    end;
            xi:=xi xor 1;yi:=yi xor 1;
            fillchar(f[yi],sizeof(f[yi]),0);
        end;
    writeln(f[xi,n,n]);
end;
 
begin
    main;
end.