【P1505 国家集训队 旅游】 树链剖分

P1505
题目描述
Ray 乐忠于旅游,这次他来到了T 城。T 城是一个水上城市,一共有 N 个景点,有些景点之间会用一座桥连接。为了方便游客到达每个景点但又为了节约成本,T 城的任意两个景点之间有且只有一条路径。换句话说, T 城中只有N − 1 座桥。

Ray 发现,有些桥上可以看到美丽的景色,让人心情愉悦,但有些桥狭窄泥泞,令人烦躁。于是,他给每座桥定义一个愉悦度w,也就是说,Ray 经过这座桥会增加w 的愉悦度,这或许是正的也可能是负的。有时,Ray 看待同一座桥的心情也会发生改变。

现在,Ray 想让你帮他计算从u 景点到v 景点能获得的总愉悦度。有时,他还想知道某段路上最美丽的桥所提供的最大愉悦度,或是某段路上最糟糕的一座桥提供的最低愉悦度。

输入格式
输入的第一行包含一个整数N,表示T 城中的景点个数。景点编号为 0…N − 1。

接下来N − 1 行,每行三个整数u、v 和w,表示有一条u 到v,使 Ray 愉悦度增加w 的桥。桥的编号为1…N − 1。|w| <= 1000。 输入的第N + 1 行包含一个整数M,表示Ray 的操作数目。

接下来有M 行,每行描述了一个操作,操作有如下五种形式:

C i w,表示Ray 对于经过第i 座桥的愉悦度变成了w。

N u v,表示Ray 对于经过景点u 到v 的路径上的每一座桥的愉悦度都变成原来的相反数。

SUM u v,表示询问从景点u 到v 所获得的总愉悦度。

MAX u v,表示询问从景点u 到v 的路径上的所有桥中某一座桥所提供的最大愉悦度。

MIN u v,表示询问从景点u 到v 的路径上的所有桥中某一座桥所提供的最小愉悦度。

测试数据保证,任意时刻,Ray 对于经过每一座桥的愉悦度的绝对值小于等于1000。

输出格式
对于每一个询问(操作S、MAX 和MIN),输出答案。

输入输出样例
不知道是不是自己树链剖分写太丑了 看别的大佬只要200行 写了第一发 (1个bug的版本写了300行
题意 就是 C 改变 x 单点值 一个单点更新 id[x]即可
N 把 x - y 区间乘 -1 我们首先对这题要知道我们进行了边权划给点权 也就是把边权值给了点中深度较大的那个点 那么我们首先要注意到 BUG 就是你需要把 最后在同一条链上的时候 Id[x]+1到y 而不是id[x] 因为你浅的点相当于不改
然后 SUM 就是求区间和
MAX 就是求区间最大值
MIN 就是求区间最小值 注意要从 id[x]+1即可 这是一个坑点

/*
    if you can't see the repay
    Why not just work step by step
    rubbish is relaxed
    to ljq
*/
#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <cmath>
#include <map>
#include <stack>
#include <set>
#include <sstream>
#include <vector>
#include <stdlib.h>
#include <algorithm>
using namespace std;

#define dbg(x) cout<<#x<<" = "<< (x)<< endl
#define dbg2(x1,x2) cout<<#x1<<" = "<<x1<<" "<<#x2<<" = "<<x2<<endl
#define dbg3(x1,x2,x3) cout<<#x1<<" = "<<x1<<" "<<#x2<<" = "<<x2<<" "<<#x3<<" = "<<x3<<endl
#define max3(a,b,c) max(a,max(b,c))
#define min3(a,b,c) min(a,min(b,c))
#define lc (rt<<1)
#define rc (rt<<11)
#define mid ((l+r)>>1)

typedef pair<int,int> pll;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int _inf = 0xc0c0c0c0;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll _INF = 0xc0c0c0c0c0c0c0c0;
const ll mod =  (int)1e9+7;

ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
ll ksm(ll a,ll b,ll mod){int ans=1;while(b){if(b&1) ans=(ans*a)%mod;a=(a*a)%mod;b>>=1;}return ans;}
ll inv2(ll a,ll mod){return ksm(a,mod-2,mod);}
void exgcd(ll a,ll b,ll &x,ll &y,ll &d){if(!b) {d = a;x = 1;y=0;}else{exgcd(b,a%b,y,x,d);y-=x*(a/b);}}//printf("%lld*a + %lld*b = %lld\n", x, y, d);
void Swap(int &x,int &y)
{
    int t = x;
    x = y;
    y = t;
}
ll Max(ll a,ll b)
{
    if(a>b) return a;
    return b;
}
ll Min(ll a,ll b)
{
    if(a<b) return a;
    return b;
}
const int MAX_N = 300025;
int n,cnt;
int p[MAX_N],eid,fa[MAX_N],sz[MAX_N],depth[MAX_N],son[MAX_N],Rank[MAX_N],top[MAX_N],id[MAX_N];
ll s[MAX_N<<2],maxx[MAX_N<<2],minn[MAX_N<<2],col[MAX_N<<2],arr[MAX_N];
void init()
{
    memset(p,-1,sizeof(p));
    eid = 0;
}
struct edge
{
    int v,next,dis;
}e[MAX_N<<1];
void add(int u,int v,int dis)
{
    e[eid].v = v;
    e[eid].dis = dis;
    e[eid].next = p[u];
    p[u] = eid++;
}
void dfs1(int x,int fath)
{
    sz[x] = 1;
    depth[x] = depth[fath] + 1;
    fa[x] = fath;
    for(int i = p[x];i+1;i=e[i].next)
    {
        int to = e[i].v;
        if(to==fath) continue;
        arr[to] = e[i].dis;
        dfs1(to,x);
        sz[x]+=sz[to];
        if(sz[to]>sz[son[x]])
            son[x] = to;
    }
}
void dfs2(int u,int t)
{
    top[u] = t;
    id[u] = ++cnt;
    Rank[cnt] = u;
    if(!son[u])
        return;
    dfs2(son[u],t);
    for(int i = p[u];i+1;i=e[i].next)
    {
        int to = e[i].v;
        if(to!=son[u]&&to!=fa[u])
            dfs2(to,to);
    }
}
void up(int rt)
{
    s[rt] = s[rt<<1] + s[rt<<1|1];
    maxx[rt] = Max(maxx[rt<<1],maxx[rt<<1|1]);
    minn[rt] = Min(minn[rt<<1],minn[rt<<1|1]);
}
void down(int rt,int l,int r)
{
    if(col[rt])
    {
        col[rt<<1] ^= 1;
        col[rt<<1|1] ^= 1;
        s[rt<<1] *= -1;
        s[rt<<1|1] *= -1;
        int t1 = maxx[rt<<1],t2 = maxx[rt<<1|1],t3 = minn[rt<<1],t4 = minn[rt<<1|1];
        maxx[rt<<1] = -t3,minn[rt<<1] = -t1,maxx[rt<<1|1] = -t4,minn[rt<<1|1] = -t2;
        col[rt] = 0;
    }
}
void build(int rt,int l,int r)
{
    if(l==r)
    {
        maxx[rt] = minn[rt] = s[rt] = arr[Rank[l]];
        return ;
    }
    build(rt<<1,l,mid);
    build(rt<<1|1,mid+1,r);
    up(rt);
}
void update(int rt,int l,int r,int x,int y)
{
    if(x>r||y<l) return;
    if(x<=l&&r<=y)
    {
        col[rt] ^= 1;
        s[rt] *= -1;
        int t = maxx[rt],s = minn[rt];
        maxx[rt] = -s,minn[rt] = -t;
        return ;
    }
    down(rt,l,r);
    if(x<=mid) update(rt<<1,l,mid,x,y);
    if(mid<y) update(rt<<1|1,mid+1,r,x,y);
    up(rt);
}
void update_(int rt,int l,int r,int x,int v)
{
    if(l==r)
    {
        s[rt] = maxx[rt] = minn[rt] = v;
        return ;
    }
    down(rt,l,r);
    if(x<=mid) update_(rt<<1,l,mid,x,v);
    else update_(rt<<1|1,mid+1,r,x,v);
    up(rt);
}
ll query(int rt,int l,int r,int x,int y)
{
    if(x>r||y<l) return 0;
    if(x<=l&&r<=y)
    {
        return s[rt];
    }
    ll res = 0;
    down(rt,l,r);
    if(x<=mid) res+=query(rt<<1,l,mid,x,y);
    if(mid<y) res+=query(rt<<1|1,mid+1,r,x,y);
    return res;
}
ll query_max(int rt,int l,int r,int x,int y)
{
    if(x>r||y<l) return -1025;
    if(x<=l&&r<=y)
    {
        return maxx[rt];
    }
    ll maxx1 = -1025,maxx2 = -1025;
    down(rt,l,r);
    if(x<=mid) maxx1 = query_max(rt<<1,l,mid,x,y);
    if(mid<y) maxx2 = query_max(rt<<1|1,mid+1,r,x,y);
    return Max(maxx1,maxx2);
}
ll query_min(int rt,int l,int r,int x,int y)
{
    if(x>r||y<l) return 1025;
    if(x<=l&&r<=y)
    {
        return minn[rt];
    }
    ll minn1 = 1025,minn2 = 1025;
    down(rt,l,r);
    if(x<=mid) minn1 = query_min(rt<<1,l,mid,x,y);
    if(mid<y) minn2 = query_min(rt<<1|1,mid+1,r,x,y);
    return Min(minn1,minn2);
}
void updates(int x,int y)
{
    while(top[x]!=top[y])
    {
        if(depth[top[x]]<depth[top[y]])
            Swap(x,y);
        update(1,1,cnt,id[top[x]],id[x]);
        x = fa[top[x]];
    }
    if(id[x]>id[y])
        Swap(x,y);
    update(1,1,cnt,id[x]+1,id[y]);
}
ll sum_max(int x,int y)
{
    ll ans = -INF;
    while(top[x]!=top[y])
    {
        if(depth[top[x]]<depth[top[y]])
            Swap(x,y);
        ans = Max(ans,query_max(1,1,n,id[top[x]],id[x]));
        x = fa[top[x]];
    }
    if(id[x]>id[y])
        Swap(x,y);
    return Max(ans,query_max(1,1,n,id[x]+1,id[y]));
}
ll sum_min(int x,int y)
{
    ll ans = INF;
    while(top[x]!=top[y])
    {
        if(depth[top[x]]<depth[top[y]])
            Swap(x,y);
        ans = Min(ans,query_min(1,1,n,id[top[x]],id[x]));
        x = fa[top[x]];
    }
    if(id[x] > id[y])
        Swap(x,y);
    return Min(ans,query_min(1,1,n,id[x]+1,id[y]));
}
ll sum(int x,int y)
{
    ll ans = 0;
    while(top[x]!=top[y])
    {
        if(depth[top[x]]<depth[top[y]])
            Swap(x,y);
        ans+=query(1,1,n,id[top[x]],id[x]);
        x = fa[top[x]];
    }
    if(id[x]>id[y])
        Swap(x,y);
    return (ans+query(1,1,n,id[x]+1,id[y]));
}
int main()
{
    //ios::sync_with_stdio(false);
    //freopen("a.txt","r",stdin);
    //freopen("b.txt","w",stdout);
    int Q,a,b,c,x,y,v;
    char str[5];
    init();
    scanf("%d",&n);
    for(int i = 1;i<n;++i)
    {
        scanf("%d%d%d",&a,&b,&c);
        a++,b++;
        add(a,b,c);
        add(b,a,c);
    }
    dfs1(1,0);
    dfs2(1,1);
    build(1,1,n);
    scanf("%d",&Q);
    while(Q--)
    {
        scanf("%s",str);
        if(str[0]=='S')
        {
            scanf("%d%d",&x,&y);x++,y++;
            printf("%lld\n",sum(x,y));
        }
        else if(str[0]=='C')
        {
            scanf("%d%d",&x,&v);x++;
            update_(1,1,n,id[x],v);
        }
        else if(str[0]=='N')
        {
            scanf("%d%d",&x,&y);x++,y++;
            updates(x,y);
        }
        else if(str[0]=='M')
        {
            scanf("%d%d",&x,&y);x++,y++;
            if(str[1]=='A')
            {
                printf("%lld\n",sum_max(x,y));
            }
            else
            {
                printf("%lld\n",sum_min(x,y));
            }
        }
    }
    //fclose(stdin);
    //fclose(stdout);
    //cout << "time: " << (long long)clock() * 1000 / CLOCKS_PER_SEC << " ms" << endl;
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值