[HAOI2012] 容易题

题目

首先,原题题目就叫这个,不是我讽刺这个题简单……
原题地址

解说

咱们先假设我一个限制都没有,那么每个数都可以取\(1~n\)之间的任何一个整数,在自己的脑海里提一遍公因数就会发现这时答案为\((1+2+ \dots +n)^m\),再用上求和公式变为\((\frac{n(n+1)}{2})^m\)
那么现在我们加上限制条件,每加上一个限制就少一个选择,那么我们就把这个限制数从这个乘数里剔除即可,即原来是\(1+2+ \dots +n\),现在变为\(1+2+ \dots +n-limit1-limit2 \dots\),也就是\((\frac{n(n+1)}{2})^m-limit1-limit2 \dots\)。而没有受到限制的数照乘不误,即若有\(js\)个有限制的数那么这些没限制的数乘起来就是\((\frac{n(n+1)}{2})^{m-js}\)。最后结果把有限制的没限制的乘一起即可。
几点提醒:
1.数太大了,能取模的地方都取模。
2.用快速幂优化。
3.样例就能看出来可能有重复限制,记得去重。

代码

#include
using namespace std;
typedef long long ll;
const ll mod=1000000007;
const int maxn=1e5+3;
map,bool> inf;
map sum;
ll n,m,k,js,num[maxn];
ll power(ll a,ll x){
	a%=mod;
	ll ans=1;
	for(;x;x>>=1,a=a*a%mod){
		if(x&1) ans=ans*a%mod;
	}
	return ans%mod;
}
ll read(){
   ll s=0,w=1;
   char ch=getchar();
   while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
   while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
   return s*w;
} 
int main(){
	n=read(); m=read(); k=read();
	while(k--){
		ll x,y;
		x=read(); y=read();
		if(!sum[x]) num[++js]=x;
		if(inf[make_pair(x,y)]) continue;
		inf[make_pair(x,y)]=1;
		sum[x]+=y;
	}
	ll ans=1,Max=(n+1)*n/2;
	for(ll i=1;i<=js;i++){
		ans*=(Max-sum[num[i]])%mod;
		ans%=mod;
	}
	printf("%lld",ans%mod*power(Max,m-js)%mod%mod);
	return 0;
}

幸甚至哉,歌以咏志。

你可能感兴趣的:([HAOI2012] 容易题)