TopCoder SRM 582 ColorfulBuilding

DP  思路是三维，但是时间肯定会超时，需要根据其特殊性质加两个标记数组，优化成二维。

刚开始想了N久N久，没感觉，还是动手画了一下才有用呀，意淫再久，不如动手呀。

代码：

#include<iostream>

#include<cstdio>

#include<vector>

#include<stack>

#include<cstring>

#include<queue>

#include<algorithm>

#include<cmath>

#include<cstring>

#include<string>

#include<set>

#include<map>



using namespace std;

const int INF=0x3f3f3f3f;

const long long MOD=1000000009;

const int N=3000;

const int M=2500;

long long dp[N][M];

long long sum[M],a[N];

int getInt(char x)

{

    if(x>='A'&&x<='Z')

    return x-'A';

    return x-'a'+26;

}

int hashChartoInt(char x,char y)

{

    return getInt(x)*52+getInt(y);

}

long long add(long long x,long long y)

{

    long long s=x+y;

    if(s>=MOD) s-=MOD;

    return s;

}

long long sub(long long x,long long y)

{

    long long s=x-y;

    if(s<=0) s+=MOD;

    return s;

}

long long mul(long long x,long long y)

{

    return (x*y)%MOD;

}

class ColorfulBuilding

{

    public:

    int count(vector <string> color1, vector <string> color2, int L)

    {

        memset(dp,0,sizeof(dp));

        memset(sum,0,sizeof(sum));

        memset(a,0,sizeof(a));

        string str1="",str2="";

        for(unsigned int i=0;i<color1.size();++i)

        str1+=color1[i];

        for(unsigned int i=0;i<color2.size();++i)

        str2+=color2[i];

        int n=str1.size();

        for(int j=0;j<N;++j)

        a[j]=1;

        for(int i=0;i<n;++i)

        {

            int nowColor=hashChartoInt(str1[n-i-1],str2[n-i-1]);

            if(i==0)

            {

                dp[nowColor][1]=1;

                sum[1]=1;

                continue;

            }

            for(int j=L;j>=1;--j)

            {

                dp[nowColor][j]=mul(dp[nowColor][j],a[nowColor]);

                sum[j]=mul(sub(sum[j],dp[nowColor][j]),i);

                dp[nowColor][j]=(dp[nowColor][j]*(i+1))%MOD;

                dp[nowColor][j]=add(dp[nowColor][j],sub(sum[j-1],mul(dp[nowColor][j-1],a[nowColor])));

                sum[j]=add(sum[j],dp[nowColor][j]);

            }

            for(int j=0;j<N;++j)

            if(j==nowColor)

            a[j]=1;

            else

            a[j]=mul(a[j],i);

        }

        return (int)(sum[L]);

    }

};