為了使得大家高興,小Q特意出個自認為的簡單題(easy)來滿足大家,這道簡單題是描述如下:有一個數列A已知對于所有的A[i]都是1~n的自然數,并且知道對于一些A[i]不能取哪些值,我們定義一個數列的積為該數列所有元素的乘積,要求你求出所有可能的數列的積的和 mod 1000000007的值,是不是很簡單呢?呵呵!
第一行三個整數n,m,k分別表示數列元素的取值范圍,數列元素個數,以及已知的限制條數。接下來k行,每行兩個正整數x,y表示A[x]的值不能是y。
一行一個整數表示所有可能的數列的積的和對1000000007取模后的結果。如果一個合法的數列都沒有,答案輸出0。
數據范圍30%的數據n<=4,m<=10,k<=10另有20%的數據k=070%的數據n<=1000,m<=1000,k<=1000100%的數據 n<=109,m<=109,k<=105,1<=y<=n,1<=x<=m
題解:快速冪+數論
答案是∏(i=1..m) ∑(j=1..n且a[i]可以等于j)j
n,m很大,k很小,所以我們將不能取到1..n中所有數的部分單獨計算,剩下的用快速冪求解即可。
#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<cmath>#include<set>#include<map>#define p 1000000007#define LL long long #define N 1000003using namespace std;int m,k,num[N],cnt;LL sum,n;map<int,LL> mp;struct data{ int x,y; bool Operator < (const data &a)const{ return (x<a.x)||(x==a.x&&y<a.y); } bool operator == (const data &a)const{ return (x==a.x&&y==a.y); }}a[N];int cmp(data a,data b){ return a.x<b.x||a.x==b.x&&a.y<b.y;}LL quickpow(LL num,int x){ LL base=num%p; LL ans=1; while (x) { if (x&1) ans=ans*base%p; x>>=1; base=base*base%p; } return ans;}int main(){ freopen("a.in","r",stdin); freopen("my.out","w",stdout); mp.clear(); scanf("%I64d%d%d",&n,&m,&k); sum=n*(n+1)%p*500000004LL%p; for (int i=1;i<=k;i++) scanf("%d%d",&a[i].x,&a[i].y); sort(a+1,a+k+1,cmp); k=unique(a+1,a+k+1)-a-1; //cout<<k<<endl; for (int i=1;i<=k;i++) { int x,y; x=a[i].x; y=a[i].y; if (mp[x]==0) num[++cnt]=x; mp[x]=(mp[x]+y)%p; if (!mp[x]) mp[x]=p; } sort(num+1,num+cnt+1); LL ans=quickpow(sum,m-cnt); //cout<<m-cnt<<endl; for (int i=1;i<=cnt;i++) { LL t=((sum-mp[num[i]])%p+p)%p; // cout<<num[i]<<" "<<mp[num[i]]<<endl; ans=(ans*t)%p; } PRintf("%I64d/n",ans);}
新聞熱點
疑難解答