[NOI2009] 管道取珠

[NOI2009] 管道取珠
非常有意思的一道dp题，状态很有意思，但卡常就是你的不对了。
题目大意是求所有的序列的重复个数的平方和。
设有k种序列，每个有$a_i$个

$$ans=\sum_{i=1}^k a_i^2$$

仔细分析发现，其实该答案就是选两次相同的序列，（第一次选该序列会有ai个答案，第二次也有ai个答案,分布乘法,两次选就是ai*ai）;
$f[i][j][k]$ 表示第一次在上部选i个，第一次在下部选j个，第二次在上部选k个，那么第二次在下部选i+j-k个；
转移具体看程序吧！；

---

代码

#include <iostream>
#include <cstdio>
using namespace std;
typedef long long ll;
const int maxn=500+1;
const int Mod=1024523;
int A[maxn],B[maxn];
int f[maxn][maxn][maxn];
int n,m;
int main()
{
    freopen("ballb.in","r",stdin);
    freopen("ballb.out","w",stdout);
    scanf("%d %d",&n,&m);
    for(int i=1; i<=n; i++)
    {
        char ch=getchar();
        while(ch!='A' && ch!='B') ch=getchar();
        ch=='A'?A[i]=1:A[i]=2;
    }
    for(int i=1; i<=m; i++)
    {
        char ch=getchar();
        while(ch!='A' && ch!='B') ch=getchar();
        ch=='A'?B[i]=1:B[i]=2;
    }
    f[0][0][0]=1;
    for(int i=0; i<=n; i++)
    {
        for(int j=0; j<=m; j++)
        {
            int tmp=n; if(i+j<=n) tmp=i+j;
            int tmp2=i+j-m; if(tmp2<0) tmp2=0;
            for(int k=tmp2; k<=tmp; k++)
            {
                int l=i+j-k;
                if(!i && !j && !k) continue;
                if(A[i]==A[k]&&i && k) f[i][j][k]+=f[i-1][j][k-1];
                if(A[i]==B[l]&&i && l) f[i][j][k]+=f[i-1][j][k];
                if(B[j]==A[k]&&j && k) f[i][j][k]+=f[i][j-1][k-1];
                if(B[j]==B[l]&&j && l) f[i][j][k]+=f[i][j-1][k];
                f[i][j][k]%=Mod;
            }
        }
    }
    cout<<f[n][m][n]<<endl;
    return 0;
}