【ZOJ】3380 Patchouli's Spell Cards

时间:2023-03-09 04:13:34
【ZOJ】3380 Patchouli's Spell Cards

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=3957

题意:m个位置,每个位置填1~n的数,求至少有L个位置的数一样的概率(1<=n,m,l<=100)

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std; struct inum {
static const int N=205, MOD=10000;
int a[N], len;
inum(int x=0) { len=!x; memset(a, 0, sizeof a); while(x) a[++len]=x%MOD, x/=MOD; }
void fix() { while(len>1 && !a[len]) --len; }
void cln() { memset(a, 0, sizeof(int)*(len+1)); len=1; }
const bool operator<= (const inum &x) {
if(len<x.len) return 1;
if(len>x.len) return 0;
for(int i=len; i; --i)
if(a[i]<x.a[i]) return 1;
else if(a[i]>x.a[i]) return 0;
return 1;
}
inum operator+ (const inum &x) {
static inum ret;
ret.cln();
ret.len=max(len, x.len);
for(int i=1; i<=ret.len; ++i) {
ret.a[i]+=a[i]+x.a[i];
if(ret.a[i]>=MOD) ret.a[i+1]+=ret.a[i]/MOD, ret.a[i]%=MOD;
}
ret.len++;
ret.fix();
return ret;
}
inum operator- (const inum &x) {
static inum ret;
ret.cln();
for(int i=1; i<=len; ++i) {
ret.a[i]+=a[i]-x.a[i];
if(ret.a[i]<0) --ret.a[i+1], ret.a[i]+=MOD;
}
ret.len=len;
ret.fix();
return ret;
}
inum operator* (const inum &x) {
static inum ret;
ret.cln();
for(int i=1; i<=len; ++i)
for(int j=1; j<=x.len; ++j) {
ret.a[i+j-1]+=a[i]*x.a[j];
if(ret.a[i+j-1]>=MOD) ret.a[i+j]+=ret.a[i+j-1]/MOD, ret.a[i+j-1]%=MOD;
}
ret.len=len+x.len;
for(int i=1; i<=ret.len; ++i)
if(ret.a[i]>=MOD) ret.a[i+1]+=ret.a[i]/MOD, ret.a[i]%=MOD;
ret.fix();
return ret;
}
inum div2() {
static inum ret;
ret.cln();
for(int i=len, t=0; i; --i) {
t+=a[i];
ret.a[i]=t>>1;
t=(t&1)*MOD;
}
ret.len=len;
ret.fix();
return ret;
}
inum operator/ (const inum &x) {
static inum l, r, mid, ONE(1);
l=0; r=*this;
while(l<=r) {
mid=(l+r).div2();
if(mid*x<=*this) l=mid+ONE;
else r=mid-ONE;
}
return l-ONE;
}
inum operator% (const inum &x) { return *this-(*this/x)*x; }
static inum gcd(inum a, inum b) { return (b.len==1&&!b.a[1])?a:gcd(b, a%b); }
static inum pow(inum a, int n) {
inum x=1;
while(n) { if(n&1) x=x*a; a=a*a; n>>=1; }
return x;
}
void P() {
printf("%d", a[len]);
for(int i=len-1; i; --i)
printf("%.04d", a[i]);
}
}; const int N=105;
inum C[N][N], d[N][N];
int n, m, l;
int main() {
C[0][0]=1;
for(int i=1; i<=100; ++i) {
C[i][0]=1;
for(int j=1; j<=i; ++j)
C[i][j]=C[i-1][j-1]+C[i-1][j];
}
while(~scanf("%d%d%d", &m, &n, &l)) {
if(l>m) { puts("mukyu~"); continue; }
d[0][0]=1; --l;
for(int i=1; i<=n; ++i)
for(int j=1; j<=m; ++j) {
d[i][j]=0; int t=min(j, l);
for(int k=0; k<=t; ++k)
d[i][j]=d[i][j]+d[i-1][j-k]*C[m-(j-k)][k];
}
inum up=0, down=0, gcd;
for(int i=1; i<=n; ++i) up=up+d[i][m];
down=inum::pow(n, m);
up=down-up;
gcd=inum::gcd(up, down);
down=down/gcd;
up=up/gcd;
up.P(); putchar('/'); down.P(); puts("");
}
return 0;
}

  

(麻麻我又码出了一个高精度模板QAQ感觉这一次封包得很好QAQ

(反正超慢QAQ差点就tle了....6000+ms....

首先本题超神!我忘记了补集转化....

求至少L个位置的数一样的方案=总方案数-少于L个位置的数一样的方案数

而后者非常好求...

设$d[i][j]$表示前$i$种数在$m$个位置上填了$j$个的方案数

$$d[i][j]=\sum_{k=0}^{min(j, L-1)} d[i-1][j-k]C[m-(j-k)][k]$$

那么$ans=\frac{n^m-\sum_{i=1}^{n} d[i][m]}{n^m}$

由于输出分数形式...于是上高精度......