Codeforces Round #588 (Div. 2)-E. Kamil and Making a Stream-求树上同一直径上两两节点之间gcd的和
【Problem Description】
给你一棵树，树上每个节点都有一个权值。定义(1sim v)的最短路径所经过的所有节点(u)称为(v)节点的祖先。定义函数(f(u,v)=gcd(u,t1,t2,dots,v))，其中(u,t1,t2,dots)都是(v)的祖先。求(sum f(u,v))。
【Solution】
对于每一个节点(v)维护一个(vector)数组，记录其所有祖先(u)对(v)的(f(u,v))的取值，以及(f(u,v))出现的次数。那么对于节点(v)的儿子节点(s)，其所有的(f(u,s))取值就为所有(f(u,v))的取值与(a[s])的(gcd)。对总答案的贡献，只要将取值乘以出现的次数即可。（其实就是很暴力的做法）
【Code】
#include <bits/stdc++.h>
using namespace std;
typedef int Int;
#define int long long
#define INF 0x3f3f3f3f
#define maxn 200000
const int mod=1e9+7;
int a[maxn];
vector<int>g[maxn];
set<int>s[maxn];
int ans=0;
int gcd(int a,int b){
return b==0?a:gcd(b,a%b);
}
map<int,int>mp[maxn];
void dfs(int u,int p){
for(auto v:g[u]){
if(v==p) continue;
for(auto vv:s[u]){ //通过父节点进行转移
int t=gcd(vv,a[v]);
(ans+=t%mod*mp[u][vv]%mod)%=mod; //贡献为取值乘以出现的次数。
mp[v][t]+=mp[u][vv]; //更新t值出现的次数
s[v].insert(t);
}
s[v].insert(a[v]);(ans+=a[v]%mod)%=mod; //最后把自己放入
mp[v][a[v]]++;
dfs(v,u);
}
}
signed main() {
ios::sync_with_stdio(false);
cin.tie(0);
int n;cin>>n;
for(int i=1;i<=n;i++) cin>>a[i];
for(int i=1;i<n;i++){
int u,v;cin>>u>>v;
g[u].push_back(v);
g[v].push_back(u);
}
s[1].insert(a[1]);ans=a[1]%mod;mp[1][a[1]]=1;
dfs(1,-1);
cout<<ans<<endl;
return 0;
}