[TC14126]BagAndCards

[TC14126]BagAndCards

题目大意：

有\(n(n\le500)\)个袋子，第\(i\)个袋子里有\(count[i][j]\)张值为\(j(j\le m\le500)\)的牌。给一个长度为\(2m-1\)的序列，告诉你\(0~2m-1\)，哪些数是好数。求对于每一对袋子\((i,j)\)，从\(i\)拿一张牌，再从\(j\)拿一张牌，两张牌之和为一个好数的方案数。

思路：

预处理\(sum[i][j]\)表示\(j\)能和\(i\)中的多少数组成好数，然后就可以直接求了。

时间复杂度\(\mathcal O(n^2m+nm^2)\)。

源代码：

#include<string>
class BagAndCards {
    private:
        using int64=long long;
        static constexpr int N=500,mod=1e9+7;
        int count[N][N],sum[N][N];
    public:
        int getHash(const int &n,const int &m,int x,const int &a,const int &b,const int &c,const std::string &isGood) {
            for(register int i=0;i<n;i++) {
                for(register int j=0;j<m;j++) {
                    count[i][j]=x;
                    x=(((int64)x*a+b)^c)%mod;
                }
            }
            for(register int i=0;i<n;i++) {
                for(register int j=0;j<m;j++) {
                    for(register int k=0;k<m;k++) {
                        if(isGood[j+k]=='Y') {
                            (sum[i][j]+=count[i][k])%=mod;
                        }
                    }
                }
            }
            int ans=0;
            for(register int i=0;i<n;i++) {
                for(register int j=i+1;j<n;j++) {
                    int tmp=0;
                    for(register int k=0;k<m;k++) {
                        (tmp+=(int64)count[i][k]*sum[j][k]%mod)%=mod;
                    }
                    ans^=tmp;
                }
            }
            return ans;
        }
};