題目鏈接：點擊查看

題目大意：給出一棵樹，規定 y 是 x 的合約數是指 y 是合數且 y 是 x 的約數，設 f[ i ] 是以 i 為根節點時，子樹內關于權值 val[ i ] 的合約數個數，求出所有的 f[ i ]?

題目分析：看到子樹問題不難想到樹啟，因為 val[ i ] 的取值范圍是 1e4，所以可以預處理出每個數字的合約數，時間復雜度是 O( nsqrt( n ) ) 的，輸出了一下發現，最多的一個數字所可以匹配的合約數最多只有 59 個，所以對于每個 val[ i ] 來說可以暴力枚舉所有的合約數，統計一下其在子樹中出現的次數即可，統計子樹中每個數字出現的次數，就是樹啟的模板題目了

代碼：
?

//#pragma GCC optimize(2) //#pragma GCC optimize("Ofast","inline","-ffast-math") //#pragma GCC target("avx,sse2,sse3,sse4,mmx") #include<iostream> #include<cstdio> #include<string> #include<ctime> #include<cmath> #include<cstring> #include<algorithm> #include<stack> #include<climits> #include<queue> #include<map> #include<set> #include<sstream> #include<cassert> #include<bitset> using namespace std;typedef long long LL;typedef unsigned long long ull;const int inf=0x3f3f3f3f;const int N=1e5+100;const int mod=1e9+7;int pri[N],a[N],num[N],son[N],cnt[N],tot;LL f[N];bool book[N],vis[N];vector<int>p[N],node[N];void P() {for(int i=2;i<N;i++){if(!book[i])pri[tot++]=i;for(int j=0;j<tot&&pri[j]*i<N;j++){book[i*pri[j]]=true;if(i%pri[j]==0)break;}} }void dfs_son(int u,int fa) {num[u]=1;son[u]=-1;for(auto v:node[u]){if(v==fa)continue;dfs_son(v,u);num[u]+=num[v];if(son[u]==-1||num[v]>num[son[u]])son[u]=v;} }void cal(int u,int fa,int val) {cnt[a[u]]+=val;for(auto v:node[u]){if(vis[v]||v==fa)continue;cal(v,u,val);} }void dfs(int u,int fa,int keep) {for(auto v:node[u]){if(v==son[u]||v==fa)continue;dfs(v,u,0);}if(son[u]!=-1){dfs(son[u],u,1);vis[son[u]]=true;}cal(u,fa,1);f[u]=0;for(auto it:p[a[u]])f[u]+=cnt[it];if(son[u]!=-1)vis[son[u]]=false;if(!keep)cal(u,fa,-1); }void init() {P();for(int i=1;i<=10000;i++)for(int j=1;j*j<=i;j++)if(i%j==0){if(book[j])p[i].push_back(j);if(j!=i/j&&book[i/j])p[i].push_back(i/j);} }int main() { #ifndef ONLINE_JUDGE // freopen("data.in.txt","r",stdin); // freopen("data.out.txt","w",stdout); #endif // ios::sync_with_stdio(false);init();int w;cin>>w;while(w--){int n,rt;scanf("%d%d",&n,&rt);memset(cnt,0,sizeof(cnt));memset(vis,false,sizeof(vis));for(int i=1;i<=n;i++)node[i].clear();for(int i=1;i<n;i++){int u,v;scanf("%d%d",&u,&v);node[u].push_back(v);node[v].push_back(u);}for(int i=1;i<=n;i++)scanf("%d",a+i);dfs_son(rt,-1);dfs(rt,-1,1);LL ans=0;for(int i=1;i<=n;i++)ans=(ans+f[i]*i)%mod;printf("%lld\n",ans);}return 0; }

?

總結

以上是生活随笔為你收集整理的牛客 - 合约数(树上启发式合并)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。