1. 程式人生 > >洛谷 P1505 [國家集訓隊]旅遊 樹鏈剖分

洛谷 P1505 [國家集訓隊]旅遊 樹鏈剖分

std pro 輸入輸出 tin pac cin 發現 chan ble

目錄

  • 題面
    • 題目鏈接
    • 題目描述
    • 輸入輸出格式
      • 輸入格式
      • 輸出格式
    • 輸入輸出樣例
      • 輸入樣例:
      • 輸出樣例:
  • 思路
  • AC代碼
  • 總結與拓展

題面

題目鏈接

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| $ \leq $ 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),輸出答案。

輸入輸出樣例

輸入樣例:

3
0 1 1
1 2 2
8
SUM 0 2
MAX 0 2
N 0 1
SUM 0 2
MIN 0 2
C 1 3
SUM 0 2
MAX 0 2

輸出樣例:

3
2
1
-1
5
3

思路

首先貌似要加個條件,好像沒給數據範圍。。。N $ \leq $ 100000 吧

題意參考如下

給定一棵樹,每條邊有一個邊權,有如下5個操作

C 修改某一條邊的權值

N 將從u到v路徑上所有邊邊權修改為其相反數

SUM 統計從u到v路徑上所有邊邊權和

MAX 統計從u到v路徑上所有邊邊權的最大值

MIN 統計從u到v路徑上所有邊邊權的最小值

操作很多,要維護一堆東西。邊權轉點權如果不懂可以先做這個題 P4114 Qtree1 這裏主要說一下線段樹的部分。

由於維護和,最大值,最小值,首先每個結構體肯定有l,r,sum,max,min;這裏C操作是單點修改邊,而N操作是區間修改,所以需要一個tag來存是否要全部變為-1,且tag初始值為1,每次N操作就要給區間的tag乘上-1。下傳時,下傳條件是t(p)==-1,sum取相反數,max是原來min的相反數,min是原來max的相反數
剩下的,碼就完了(這是一道debug題)

AC代碼

#include<bits/stdc++.h>
const int maxn=100010;
using namespace std;

int n,m,w[maxn];
int tot,wt[maxn],to[maxn<<1],nxt[maxn<<1],head[maxn];
int son[maxn],fa[maxn],dep[maxn],len[maxn];
int cnt,top[maxn],nid[maxn],nw[maxn];
struct BuildTree
{
    int l,r,mx,mi,sum,tag;
    #define l(a) tree[a].l
    #define r(a) tree[a].r
    #define m(a) ((l(a)+r(a))>>1)
    #define s(a) tree[a].sum
    #define t(a) tree[a].tag
    #define mx(a) tree[a].mx
    #define mi(a) tree[a].mi
}tree[maxn<<2];

int read() ///又看到0開頭,逼我學快讀
{
    int lsk=0,f=1;
    char ch=getchar();
    while(!isdigit(ch)) {if(ch==‘-‘) f=-1;ch=getchar();}
    while(isdigit(ch)) lsk=(lsk<<3)+(lsk<<1)+(ch-‘0‘),ch=getchar();
    return f*lsk;
}

void dfs1(int u,int f,int d)
{
    fa[u]=f;dep[u]=d;len[u]=1;
    for(int i=head[u];i;i=nxt[i])
    {
        int v=to[i];
        if(v==f) continue;
        dfs1(v,u,d+1);
        len[u]+=len[v];
        if(len[v]>len[son[u]]) son[u]=v;
    }
}

void dfs2(int p,int t)
{
    top[p]=t;nid[p]=++cnt;
    if(!son[p]) return;
    dfs2(son[p],t);
    for(int i=head[p];i;i=nxt[i])
    {
        int v=to[i];
        if(v==fa[p] || v==son[p]) continue;
        dfs2(v,v);
    }
}

void Update(int p)
{
    s(p)=s(p<<1)+s(p<<1|1);
    mx(p)=max(mx(p<<1),mx(p<<1|1));
    mi(p)=min(mi(p<<1),mi(p<<1|1));
}

void BuildTree(int p,int l,int r)
{
    l(p)=l;r(p)=r;t(p)=1;
    if(l==r)
    {
        s(p)=mx(p)=mi(p)=nw[l];
        return;
    }
    BuildTree(p<<1,l,m(p));
    BuildTree(p<<1|1,m(p)+1,r);
    Update(p);
}

void R(int p)
{
    s(p)=-s(p);
    swap(mi(p),mx(p));
    mi(p)=-mi(p);
    mx(p)=-mx(p);
    t(p)*=-1;
}

void PushDown(int p)
{
    if(t(p)==-1)
    {
        R(p<<1);
        R(p<<1|1);
        t(p)=1;
    }
}

void SC(int np,int p,int k) ///SingleChange
{
    if(l(np)==r(np))
    {
        s(np)=mx(np)=mi(np)=k;
        return;
    }
    PushDown(np);
    if(p<=m(np)) SC(np<<1,p,k);
    if(p>m(np)) SC(np<<1|1,p,k);
    Update(np);
}

void RC1(int p,int l,int r) ///ReverseChange
{
    if(l<=l(p) && r>=r(p))
    {
        R(p);
        return;
    }
    PushDown(p);
    if(l<=m(p)) RC1(p<<1,l,r);
    if(r>m(p)) RC1(p<<1|1,l,r);
    Update(p);
}

void RC2(int u,int v) 
{
    while(top[u]!=top[v])
    {
        if(dep[top[u]]<dep[top[v]]) swap(u,v);
        RC1(1,nid[top[u]],nid[u]);
        u=fa[top[u]];
    }
    if(u==v) return;
    if(dep[u]>dep[v]) swap(u,v);
    RC1(1,nid[u]+1,nid[v]);
}

int GS1(int p,int l,int r) ///GetSum
{
    if(l<=l(p) && r>=r(p)) return s(p);
    PushDown(p);
    int ans=0;
    if(l<=m(p)) ans+=GS1(p<<1,l,r);
    if(r>m(p)) ans+=GS1(p<<1|1,l,r);
    return ans;
}

int GS2(int u,int v)
{
    int ans=0;
    while(top[u]!=top[v])
    {
        if(dep[top[u]]<dep[top[v]]) swap(u,v);
        ans+=GS1(1,nid[top[u]],nid[u]);
        u=fa[top[u]];
    }
    if(u==v) return ans;
    if(dep[u]>dep[v]) swap(u,v);
    ans+=GS1(1,nid[u]+1,nid[v]);
    return ans;
}

int GMX1(int p,int l,int r) ///GetMax
{
    if(l<=l(p) && r>=r(p)) return mx(p);
    PushDown(p);
    int ans=INT_MIN;
    if(l<=m(p)) ans=max(ans,GMX1(p<<1,l,r));
    if(r>m(p)) ans=max(ans,GMX1(p<<1|1,l,r));
    return ans;
}

int GMX2(int u,int v)
{
    int ans=INT_MIN;
    while(top[u]!=top[v])
    {
        if(dep[top[u]]<dep[top[v]]) swap(u,v);
        ans=max(ans,GMX1(1,nid[top[u]],nid[u]));
        u=fa[top[u]];
    }
    if(u==v) return ans;
    if(dep[u]>dep[v]) swap(u,v);
    ans=max(ans,GMX1(1,nid[u]+1,nid[v]));
    return ans;
}

int GMI1(int p,int l,int r) ///GetMin
{
    if(l<=l(p) && r>=r(p)) return mi(p);
    PushDown(p);
    int ans=INT_MAX;
    if(l<=m(p)) ans=min(ans,GMI1(p<<1,l,r));
    if(r>m(p)) ans=min(ans,GMI1(p<<1|1,l,r));
    return ans;
}

int GMI2(int u,int v)
{
    int ans=INT_MAX;
    while(top[u]!=top[v])
    {
        if(dep[top[u]]<dep[top[v]]) swap(u,v);
        ans=min(ans,GMI1(1,nid[top[u]],nid[u]));
        u=fa[top[u]];
    }
    if(u==v) return ans;
    if(dep[u]>dep[v]) swap(u,v);
    ans=min(ans,GMI1(1,nid[u]+1,nid[v]));
    return ans;
}

int main()
{
    n=read();
    for(int i=1;i<n;i++)
    {
        int u,v,w;
        u=read()+1;
        v=read()+1;
        w=read();wt[i]=w;
        to[++tot]=v;nxt[tot]=head[u];head[u]=tot;
        to[++tot]=u;nxt[tot]=head[v];head[v]=tot;
    }
    dfs1(1,1,1);
    dfs2(1,1);
    for(int i=1;i<n;i++)
    {
        int u=to[(i<<1)-1],v=to[i<<1];
        if(dep[u]<dep[v]) nw[nid[v]]=wt[i];
        else nw[nid[u]]=wt[i];
    }
    BuildTree(1,1,n);
    m=read();
    for(int i=1;i<=m;i++)
    {
        string way;int u,v;
        cin>>way;u=read()+1;v=read()+1;
        if(way=="C")
        {
            u--,v--;
            int a=to[(u<<1)-1],b=to[u<<1];
            if(dep[a]<dep[b]) SC(1,nid[b],v);
            else SC(1,nid[a],v);
        }
        else if(way=="N") RC2(u,v);
        else if(way=="SUM") printf("%d\n",GS2(u,v));
        else if(way=="MAX") printf("%d\n",GMX2(u,v));
        else printf("%d\n",GMI2(u,v));
    }
    return 0;
}

總結與拓展

碼量很大的題目,ctrl+c別忘了改其中的東西

洛谷 P1505 [國家集訓隊]旅遊 樹鏈剖分