hihocoder#1347 : 小h的树上的朋友（树链剖分）

时间限制: 18000ms

单点时限: 2000ms

内存限制: 512MB

描述

小h拥有n位朋友。每位朋友拥有一个数值Vi代表他与小h的亲密度。亲密度有可能发生变化。

岁月流逝，小h的朋友们形成了一种稳定的树状关系。每位朋友恰好对应树上的一个节点。

每次小h想请两位朋友一起聚餐，他都必须把连接两位朋友的路径上的所有朋友都一起邀请上。并且聚餐的花费是这条路径上所有朋友的亲密度乘积。

小h很苦恼，他需要知道每一次聚餐的花销。小h问小y，小y当然会了，他想考考你。

输入

输入文件第一行是一个整数n，表示朋友的数目，从1开始编号。

输入文件第二行是n个正整数Vi，表示每位朋友的初始的亲密度。

接下来n-1行，每行两个整数u和v，表示u和v有一条边。

然后是一个整数m，代表操作的数目。每次操作为两者之一：

0 u v 询问邀请朋友u和v聚餐的花费

1 u v 改变朋友u的亲密度为v

1<=n,m<=5*105

Vi<=109

输出

对于每一次询问操作，你需要输出一个整数，表示聚餐所需的花费。你的答案应该模1,000,000,007输出。

样例输入

3
1 2 3
1 2
2 3
5
0 1 2
0 1 3
1 2 3
1 3 5
0 1 3

样例输出

2
6
15

思路：树链剖分模版题。

#include<bits/stdc++.h>
using namespace std;
const int MAX=1e6;
const int MOD=1e9+7;
typedef long long ll;
vector<int>e[MAX];
int a[MAX],d[MAX],L[MAX],fa[MAX],siz[MAX],son[MAX],tp[MAX],val[MAX],all;
void dfs1(int k,int f,int dep)
{
    d[k]=dep;
    siz[k]=1;
    son[k]=0;
    fa[k]=f;
    for(int i=0;i<e[k].size();i++)
    {
        int nex=e[k][i];
        if(nex==f)continue;
        dfs1(nex,k,dep+1);
        siz[k]+=siz[nex];
        if(siz[son[k]]<siz[nex])son[k]=nex;
    }
}
void dfs2(int k,int top)
{
    tp[k]=top;
    L[k]=++all;
    val[all]=a[k];
    if(son[k])dfs2(son[k],top);
    for(int i=0;i<e[k].size();i++)
    {
        int nex=e[k][i];
        if(nex==son[k]||nex==fa[k])continue;
        dfs2(nex,nex);
    }
}
struct lenka
{
    int l,r;
    ll ans;
}A[MAX<<2];
void build(int k,int  l,int r)
{
    A[k].l=l,A[k].r=r;
    if(l==r){A[k].ans=val[l]%MOD;return;}
    build(2*k,l,(l+r)/2);
    build(2*k+1,(l+r)/2+1,r);
    A[k].ans=A[2*k].ans*A[2*k+1].ans%MOD;
}
void change(int k,int x,ll y)
{
    if(x==A[k].l&&x==A[k].r){A[k].ans=y%MOD;return;}
    if(x<=A[2*k].r)change(2*k,x,y);
    else change(2*k+1,x,y);
    A[k].ans=A[2*k].ans*A[2*k+1].ans%MOD;
}
ll ask(int k,int x,int y)
{
    if(x==A[k].l&&y==A[k].r)return A[k].ans;
    if(y<=A[2*k].r)return ask(2*k,x,y);
    else if(x>=A[2*k+1].l)return ask(2*k+1,x,y);
    return ask(2*k,x,A[2*k].r)*ask(2*k+1,A[2*k+1].l,y)%MOD;
}
ll get(int x,int y)
{
    ll ans=1;
    while(tp[x]!=tp[y])
    {
        if(d[tp[x]]<d[tp[y]])swap(x,y);
        ans=ans*ask(1,L[tp[x]],L[x])%MOD;
        x=fa[tp[x]];
    }
    if(d[x]>d[y])swap(x,y);
    return ans*ask(1,L[x],L[y])%MOD;
}
int main()
{
    int n,m;
    cin>>n;
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        e[x].push_back(y);
        e[y].push_back(x);
    }
    dfs1(1,1,1);
    all=0;
    dfs2(1,1);
    build(1,1,all);
    cin>>m;
    while(m--)
    {
        int op,x,y;
        scanf("%d%d%d",&op,&x,&y);
        if(op==1)change(1,L[x],y);
        else printf("%lld\n",get(x,y));
    }
    return 0;
}