首页 > 代码库 > hdu4390-Number Sequence(容斥计算)
hdu4390-Number Sequence(容斥计算)
题意:给定b数列,计算有多少种数列 a1,a2,...,an 满足条件 a1*a2*...*an=b1*b2*…*bn (ai>1).
解法:处理出b数列中出现的所有质因子的数量记录在map中,然后进行容斥计算:
代码:
/****************************************************** * author:xiefubao *******************************************************/ #pragma comment(linker, "/STACK:102400000,102400000") #include <iostream> #include <cstring> #include <cstdlib> #include <cstdio> #include <queue> #include <vector> #include <algorithm> #include <cmath> #include <map> #include <set> #include <stack> #include <string.h> //freopen ("in.txt" , "r" , stdin); using namespace std; #define eps 1e-8 #define zero(_) (abs(_)<=eps) const double pi=acos(-1.0); typedef long long LL; const int Max=1000010; const int INF=1000000007; map<int,int> maps; int num[30]; int n; bool rem[1000010]; LL C[500][500]; int prime[Max/2]; int p=0; void init() { for(LL i=2; i<Max; i++) if(!rem[i]) { prime[p++]=i; for(LL j=i*i; j<Max; j+=i) rem[j]=1; } for(int i=0; i<500; i++) for(int j=0; j<=i; j++) C[i][j]=j?(C[i-1][j-1]+C[i-1][j])%INF:1; } void make(int t) { int tool=t; for(int i=0; prime[i]*prime[i]<=t; i++) { while(tool%prime[i]==0) maps[prime[i]]++,tool/=prime[i]; } if(tool>=2) maps[tool]++; } LL getans(int m) { LL ans=1; int k=n-m; for(map<int,int>::iterator p=maps.begin();p!=maps.end();p++) { int t=p->second; ans=(ans*C[k+t-1][k-1])%INF; } return ans; } int main() { init(); while(scanf("%d",&n)==1) { maps.clear(); for(int i=0; i<n; i++) scanf("%d",num+i); for(int i=0; i<n; i++) { make(num[i]); } int ans=0; for(int i=0; i<n; i++) { if(i&1) { ans-=C[n][i]*getans(i)%INF; if(ans<0) ans+=INF; } else { ans+=C[n][i]*getans(i)%INF; if(ans>=INF) ans-=INF; } } cout<<ans<<'\n'; } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。