【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}$

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