树剖模板

#include<cstdio>
#include<cctype>
#include<cstring>
#include<algorithm>
using namespace std;
inline long long read()
{
    long long x = 0, flag = 1;
    char c;
    while(! isdigit(c = getchar()))
        if(c == '-')
            flag *= - 1;
    while(isdigit(c)) 
        x = x * 10 + c - '0', c = getchar();
    return x * flag;
}
const long long MAXN = (long long)1e5 + (1 << 5);
long long w[MAXN];
long long top;
struct edge
{
    long long v, next;
}G[MAXN << 1];
long long head[MAXN];
long long n, m, root, MOD;
void add_edge(long long u, long long v)
{
    G[top].v = v;
    G[top].next = head[u];
    head[u] = top ++;
}
struct node
{
    long long w, dep, fa, son, size, top, ID, last;   
}a[MAXN];
void DFS1(long long u, long long fa)
{
    a[u].dep = a[fa].dep + 1;
    a[u].fa = fa;
    a[u].size = 1;
    a[u].son = - 1;
    for(long long i = head[u]; i != - 1; i = G[i].next)
    {
        long long v = G[i].v;
        if(v == fa)
            continue;
        DFS1(v, u);
        a[u].size += a[v].size;
        if(a[u].son == - 1 || a[v].size > a[a[u].son].size)
            a[u].son = v;
    }
}
long long cnt;
void DFS2(long long u, long long top)
{
    a[u].top = top;
    a[u].ID = ++ cnt;
    if(a[u].son == - 1)
    {
        a[u].last = a[u].ID;
        return;
    }
    DFS2(a[u].son, top);
    for(long long i = head[u]; i != - 1; i = G[i].next)
        if(G[i].v != a[u].son && G[i].v != a[u].fa)
            DFS2(G[i].v, G[i].v);
    a[u].last = cnt;
}
struct segmenttree
{
    long long sum, tag;
}T[MAXN << 2]; 
void modify(long long u, long long curL, long long curR, long long L, long long R, long long delta)
{
    if(curL >= L && curR <= R)
    {
        T[u].tag += delta;
        T[u].sum += delta * (curR - curL + 1);
        return; 
    }
    long long mid = (curL + curR) >> 1;
    if(L <= mid)
        modify(u << 1, curL, mid, L, R, delta);
    if(R > mid)
        modify((u << 1) + 1, mid + 1, curR, L, R, delta);
    T[u].sum += (min(curR, R) - max(curL, L) + 1) * delta;
}
void update(long long u, long long v, long long delta)
{
    while(a[u].top != a[v].top)
    {
        if(a[a[u].top].dep < a[a[v].top].dep)
            swap(u, v);
        modify(1, 1, n, a[a[u].top].ID, a[u].ID, delta);
        u = a[a[u].top].fa; 
    }
    if(a[u].ID < a[v].ID)
        modify(1, 1, n, a[u].ID, a[v].ID, delta);
    else
        modify(1, 1, n, a[v].ID, a[u].ID, delta);
}
long long query(long long u, long long curL, long long curR, long long L, long long R)
{
    if(curL >= L && curR <= R)
        return T[u].sum;
    long long mid = (curL + curR) >> 1, ret = 0;
    if(L <= mid)
        ret += query(u << 1, curL, mid, L, R);
    if(R > mid)
        ret += query((u << 1) + 1, mid + 1, curR, L, R);
    ret += (min(curR, R) - max(curL, L) + 1) * T[u].tag;
    return ret;
}
long long ask(long long u, long long v)
{
    long long ret = 0;
    while(a[u].top != a[v].top)
    {
        if(a[a[u].top].dep < a[a[v].top].dep)
            swap(u, v);
        ret += query(1, 1, n, a[a[u].top].ID, a[u].ID);
        u = a[a[u].top].fa;
    }
    if(a[u].ID < a[v].ID)
        ret += query(1, 1, n, a[u].ID, a[v].ID);
    else
        ret += query(1, 1, n, a[v].ID, a[u].ID);
    return ret;
}
void print(long long x)
{
    if(x < 0)
        putchar('-');
    if(x == 0)
        putchar('0');
    long long top = 0, ans[36];
    while(x)
        ans[top ++] = x % 10, x /= 10;
    for(; top; top --)
        putchar(ans[top - 1] + '0');
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("L3384.in", "r", stdin);
    freopen("L3384.out", "w", stdout);
    #endif
    n = read(), m = read(), root = read(), MOD = read();
    for(long long i = 1; i <= n; i ++)
        a[i].w = read();
    top = 0;
    memset(head, - 1, sizeof(head));
    for(long long i = 1; i < n; i ++)
    {
        long long u = read(), v = read();
        add_edge(u, v);
        add_edge(v, u);
    }
    a[root].dep = - 1;
    DFS1(root, root);
    cnt = 0;
    DFS2(root, root);
    memset(T, 0, sizeof(T)); 
    for(long long i = 1; i <= n; i ++)
        modify(1, 1, n, a[i].ID, a[i].ID, a[i].w);
    for(long long i = 0; i < m; i ++)
    {
        long long opt = read();
        switch (opt)
        {
            case 1:
                {
                    long long u = read(), v = read(), delta = read();
                    update(u, v, delta);
                    break;
                }
            case 2:
                {
                    long long u = read(), v = read();
                    print(ask(u, v) % MOD);
                    putchar('\n');
                    break;
                }
            case 3:
                {
                    long long u = read(), delta = read();
                    modify(1, 1, n, a[u].ID, a[u].last, delta);
                    break;
                }
            case 4:
                {
                    long long u = read();
                    print(query(1, 1, n, a[u].ID, a[u].last) % MOD);
                    putchar('\n');
                    break;
                }
        }
    }
}
#include <cstdio> #include <iostream> #include <vector> #define N 30003 #define INF 2147483647 using namespace std; int n,f[N][20],dep[N],siz[N],son[N],top[N],tot,pos[N],w[N]; int Max[N*4],Sum[N*4]; vector <int> to[N]; void dfs1(int x){ siz[x]=1; int sz=to[x].size(); for(int i=0;i<sz;++i){ int y=to[x][i]; if(y==f[x][0])continue; f[y][0]=x; dep[y]=dep[x]+1; dfs1(y); siz[x]+=siz[y]; if(siz[y]>siz[son[x]])son[x]=y; } } void dfs2(int x,int root){ top[x]=root; pos[x]=++tot; if(son[x])dfs2(son[x],root); int sz=to[x].size(); for(int i=0;i<sz;++i){ int y=to[x][i]; if(y==f[x][0] || y==son[x])continue; dfs2(y,y); } } void update(int k,int l,int r,int P,int V){ if(l==r){ Max[k]=Sum[k]=V; return; } int mid=(l+r)>>1; if(P<=mid)update(k*2,l,mid,P,V); else update(k*2+1,mid+1,r,P,V); Max[k]=max(Max[k*2],Max[k*2+1]); Sum[k]=Sum[k*2]+Sum[k*2+1]; } void up(int &x,int goal){ for(int i=15;i>=0;--i) if(dep[f[x][i]]>=goal)x=f[x][i]; } int lca(int x,int y){ if(dep[x]>dep[y])up(x,dep[y]); if(dep[x]<dep[y])up(y,dep[x]); if(x==y)return x; for(int i=15;i>=0;--i) if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i]; return f[x][0]; } int getm(int k,int l,int r,int L,int R){ if(L<=l && r<=R)return Max[k]; int res=-INF,mid=(l+r)>>1; if(L<=mid)res=max(res,getm(k*2,l,mid,L,R)); if(R>mid)res=max(res,getm(k*2+1,mid+1,r,L,R)); return res; } int gets(int k,int l,int r,int L,int R){ if(L<=l && r<=R)return Sum[k]; int res=0,mid=(l+r)>>1; if(L<=mid)res+=gets(k*2,l,mid,L,R); if(R>mid)res+=gets(k*2+1,mid+1,r,L,R); return res; } int main(){ scanf("%d",&n); for(int i=1,a,b;i<n;++i){ scanf("%d%d",&a,&b); to[a].push_back(b); to[b].push_back(a); } dep[1]=1; dfs1(1); dfs2(1,1); for(int i=1;i<=15;++i) for(int j=1;j<=n;++j)f[j][i]=f[f[j][i-1]][i-1]; for(int i=1;i<=n;++i){ scanf("%d",&w[i]); update(1,1,n,pos[i],w[i]); } int q; scanf("%d",&q); while(q--){ char s[10]; int u,v,t; scanf("%s",s); if(s[1]=='H'){ scanf("%d%d",&u,&t); w[u]=t; update(1,1,n,pos[u],t); } if(s[1]=='M'){ scanf("%d%d",&u,&v); int ans=-INF,t=lca(u,v); for(int i=u;i;i=f[top[i]][0]) if(dep[t]<dep[top[i]]) ans=max(ans,getm(1,1,n,pos[top[i]],pos[i])); else{ ans=max(ans,getm(1,1,n,pos[t],pos[i])); break; } for(int i=v;i;i=f[top[i]][0]) if(dep[t]<dep[top[i]]) ans=max(ans,getm(1,1,n,pos[top[i]],pos[i])); else{ ans=max(ans,getm(1,1,n,pos[t],pos[i])); break; } printf("%d\n",ans); } if(s[1]=='S'){ scanf("%d%d",&u,&v); int ans=0,t=lca(u,v); for(int i=u;i;i=f[top[i]][0]) if(dep[t]<dep[top[i]]) ans+=gets(1,1,n,pos[top[i]],pos[i]); else{ ans+=gets(1,1,n,pos[t],pos[i]); break; } for(int i=v;i;i=f[top[i]][0]) if(dep[t]<dep[top[i]]) ans+=gets(1,1,n,pos[top[i]],pos[i]); else{ ans+=gets(1,1,n,pos[t],pos[i]); break; } printf("%d\n",ans-w[t]); } } }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值