BZOJ 1566 파이프 구슬 뽑기(DP 좋은 문제)

제목: 왼쪽에 상하 두 개의 파이프가 있는데 각각 n개와 m개의 다른 색깔의 두 개의 공이 있고 오른쪽에 빈 파이프가 있다.매번 왼쪽의 파이프를 선택하여 가장 오른쪽의 공을 오른쪽 파이프로 밀어내고 n+m회 조작을 거쳐 오른쪽 파이프는 오른쪽에서 왼쪽으로 출력 서열을 형성한다.서로 다른 종류의 출력 서열의 생성 방식수의 제곱의 합을 구하다.n,m <= 500
사고방식: 이 제곱은 보기에 매우 어렵다.교묘한 전환이 하나 있는데, 우리는 두 사람이 이 게임을 한다고 볼 수 있다. 그러면 답은 두 번째 사람의 것이다
모든 방안이 첫 번째 사람의 모든 방안에 나타난 횟수의 합.이렇게 하면 DP가 할 수 있습니다. dp[i][j][k]는 i개를 얻었고 첫 번째 사람이 위에서 가져왔습니다.
j개, 두 번째 사람이 위에서 k개의 같은 방안의 방안 수를 가져왔다.전송 코드
코드:

#include
using namespace std;
const int mod = 1024523;
const int maxn = 505;
int dp[2][maxn][maxn];
int n, m;
char str1[maxn], str2[maxn];

int main(void)
{
    while(cin >> n >> m)
    {
        scanf(" %s %s", str1, str2);
        for(int i = 0; i < n/2; i++) swap(str1[i], str1[n-i-1]);
        for(int i = 0; i < m/2; i++) swap(str2[i], str2[m-i-1]);
        memset(dp, 0, sizeof(dp));
        dp[0][0][0] = 1;
        int cur = 0, pre;
        for(int i = 0; i < n+m; i++)
        {
            pre = cur;
            cur = !pre;
            memset(dp[cur], 0, sizeof(dp[cur]));
            int l = max(i-m, 0);
            int r = min(i, n);
            for(int up1 = l; up1 <= r; up1++)
            {
                for(int up2 = l; up2 <= r; up2++)
                {
                    if(!dp[pre][up1][up2]) continue;
                    int down1 = i-up1;
                    int down2 = i-up2;
                    //   
                    if(str1[up1] == str1[up2]) dp[cur][up1+1][up2+1] = (dp[cur][up1+1][up2+1]+dp[pre][up1][up2])%mod;
                    //p1 p2 
                    if(str1[up1] == str2[down2]) dp[cur][up1+1][up2] = (dp[cur][up1+1][up2]+dp[pre][up1][up2])%mod;
                    //   
                    if(str2[down1] == str2[down2]) dp[cur][up1][up2] = (dp[cur][up1][up2]+dp[pre][up1][up2])%mod;
                    //p1 p2 
                    if(str2[down1] == str1[up2]) dp[cur][up1][up2+1] = (dp[cur][up1][up2+1]+dp[pre][up1][up2])%mod;
                }
            }
        }
        printf("%d
", dp[cur][n][n]);
    }
    return 0;
}