【CF1210C】Kamil and Making a Stream（vector，数论，树）

题意：给定一棵n个点带点权的树，i号点的点定义f(i,j)为i到j路径上所有点的gcd，其中i是j的一个祖先，求所有f(i,j)之和mod1e9+7

2<=n<=1e5，0<=a[i]<=1e12

思路：从根往下直接暴力跑，每个点开个vector继承父节点的信息，取gcd之后再把值相等的合并

大胆猜想，不用求证

注意0还是要保留的

 #include<bits/stdc++.h>
 using namespace std;
 typedef long long ll;
 typedef unsigned int uint;
 typedef unsigned long long ull;
 typedef pair<int,int> PII;
 typedef pair<ll,ll> Pll;
 typedef vector<int> VI;
 typedef vector<PII> VII;
 typedef pair<ll,ll>P;
 #define N  200010
 #define M  200010
 #define fi first
 #define se second
 #define MP make_pair
 #define pi acos(-1)
 #define mem(a,b) memset(a,b,sizeof(a))
 #define rep(i,a,b) for(int i=(int)a;i<=(int)b;i++)
 #define per(i,a,b) for(int i=(int)a;i>=(int)b;i--)
 #define lowbit(x) x&(-x)
 #define Rand (rand()*(1<<16)+rand())
 #define id(x) ((x)<=B?(x):m-n/(x)+1)
 #define ls p<<1
 #define rs p<<1|1

 const ll MOD=1e9+,inv2=(MOD+)/;
       double eps=1e-;
       int INF=<<;
       ll inf=5e13;
       int dx[]={-,,,};
       int dy[]={,,-,};

 int head[N],vet[N],nxt[N],fa[N],d[N],vis[N],tot;
 ll a[N],ans;
 P b[N];
 vector<P> c[N];

 int read()
 {
    int v=,f=;
    char c=getchar();
    while(c<||<c) {if(c=='-') f=-; c=getchar();}
    while(<=c&&c<=) v=(v<<)+v+v+c-,c=getchar();
    return v*f;
 }

 void add(int a,int b)
 {
     nxt[++tot]=head[a];
     vet[tot]=b;
     head[a]=tot;
 }

 ll gcd(ll x,ll y)
 {
     if(!y) return x;
     return gcd(y,x%y);
 }

 void dfs(int u,int fa)
 {
     //printf("u=%d fa=%d\n",u,fa);
     if(fa)
     {
         for(int i=;i<c[fa].size();i++) c[u].push_back(c[fa][i]);
         //c[u]=c[fa];
     }

     for(int i=;i<c[u].size();i++) c[u][i].fi=gcd(c[u][i].fi,a[u]);
     int m=;
     for(int i=;i<c[u].size();i++) b[++m]=c[u][i];
     sort(b+,b+m+);
     c[u].clear();
     rep(i,,m)
     {
         int now=(int)c[u].size()-;
         if(now==-||b[i].fi!=c[u][now].fi) c[u].push_back(b[i]);
          else c[u][now].se=(c[u][now].se+b[i].se)%MOD;
     }
     c[u].push_back(MP(a[u],));
     for(int i=;i<c[u].size();i++) ans=(ans+c[u][i].fi%MOD*c[u][i].se%MOD)%MOD;
     int e=head[u];
     while(e)
     {
         int v=vet[e];
         if(v!=fa) dfs(v,u);
         e=nxt[e];
     }
     //printf("u=%d fa=%d ans=%I64d\n",u,fa,ans);
 }

 int main()
 {
     //freopen("1.in","r",stdin);
     int n=read();
     rep(i,,n) scanf("%I64d",&a[i]);
     tot=;
     rep(i,,n) head[i]=;
     rep(i,,n-)
     {
         int x=read(),y=read();
         add(x,y);
         add(y,x);
     }
     dfs(,);
     printf("%I64d\n",ans);
     return ;
 }