[TC14126]BagAndCards
题目大意:
有(n(nle500))个袋子,第(i)个袋子里有(count[i][j])张值为(j(jle mle500))的牌。给一个长度为(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;
}
};