bzoj1566 [NOI2009]管道取珠

题目链接：bzoj1566
题目大意：
两个输入管道里面分别有n、m个有着两种颜色的珠子。只从右端取。依次于输出管道中形成。设对于一种的输出方式有$a_i$种产生方法有K种不同的输出方式，求$\sum_{i=1}^Ka_i^2$。

题解：
智商DP
看到$a_i^2$想到什么呢？$a_i\times a_i$
嗯..$a_i$是相同方式的产生方法的个数，那么$a_i\times a_i$就可以看成两个人来玩这个游戏，产生方法相同的个数。
可是我想不到啊
设$f[i][j][k]$表示要取第$i$个球，第一个人在上方管道取了$j$个，第二个人在上方管道取了 k <script type="math/tex" id="MathJax-Element-1369">k</script>个。
转移就很简单了

但是我还是在奇怪的地方纠结了好久qwq

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define N 510

const int mod=1024523;
int a[N],b[N],f[2][N][N];char s[N];
int main()
{
    //freopen("a.in","r",stdin);
    //freopen("a.out","w",stdout);
    int t=0,n,m,i,j,k;
    scanf("%d%d\n",&n,&m);
    gets(s+1);
    for (i=1;i<=n;i++) a[i]=s[i]-'A';
    gets(s+1);
    for (i=1;i<=m;i++) b[i]=s[i]-'A';
    f[0][0][0]=1;t=1;
    for (i=1;i<=n+m;i++)
    {
        for (j=0;j<=n;j++)//x从上面取了j
         for (k=0;k<=n;k++)//y从上面取了k
          if (f[1-t][j][k]>0)
          {
             int tt=f[1-t][j][k];
             int ax=n-j,ay=n-k,bx=m-(i-1-j),by=m-(i-1-k);
             if (ax>0 && ay>0 && a[ax]==a[ay]) f[t][j+1][k+1]=(f[t][j+1][k+1]+tt)%mod;
             if (ax>0 && by>0 && a[ax]==b[by]) f[t][j+1][k]=(f[t][j+1][k]+tt)%mod;
             if (bx>0 && ay>0 && b[bx]==a[ay]) f[t][j][k+1]=(f[t][j][k+1]+tt)%mod;
             if (bx>0 && by>0 && b[bx]==b[by]) f[t][j][k]=(f[t][j][k]+tt)%mod;
          }
        t=1-t;
        memset(f[t],0,sizeof(f[t]));
    }
    printf("%d\n",f[1-t][n][n]);
    return 0;
}