2021江西省赛 I. Homework（树形dp换根）

Posted 2021-11-21 live4m

题意：

解法：

由于查询只有200次,我们在op=3的时候再计算答案即可,不需要每次操作都重新计算答案.

一开始上当了,以为要从完成时间最小的开始递推,用堆存(val,idx)一直取,
复杂度O(n*log*200),题目时限4s以为能够,结果t了.

发现对于每个点x,暴力计算答案的话就是枚举点v,d[x]=min(d[x],d[v]+w),w是路径长度.
可以树形dp换根做,复杂度就降到O(n)了.

ps:
这题不能用vector存图,卡常.

code：

#include<bits/stdc++.h>
using namespace std;
const int maxm=2e5+5;
int head[maxm],nt[maxm<<1],to[maxm<<1],idx[maxm<<1],cnt;
int d[maxm];
int a[maxm];
int e[maxm];
int n,q;
void add(int x,int y,int z){
    cnt++;nt[cnt]=head[x];head[x]=cnt;to[cnt]=y;idx[cnt]=z;
}
void dfs1(int x,int fa){
    d[x]=a[x];
    for(int i=head[x];i;i=nt[i]){
        int v=to[i],w=e[idx[i]];
        if(v==fa)continue;
        dfs1(v,x);
        d[x]=min(d[x],d[v]+w);
    }
}
void dfs2(int x,int fa){
    for(int i=head[x];i;i=nt[i]){
        int v=to[i],w=e[idx[i]];
        if(v==fa)continue;
        d[v]=min(d[v],d[x]+w);
        dfs2(v,x);
    }
}
void cal(){
    dfs1(1,1);
    dfs2(1,1);
    int ans=0;
    for(int i=1;i<=n;i++){
        ans^=d[i];
    }
    printf("%d\\n",ans);
}
void solve(){
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    for(int i=1;i<n;i++){
        int x,y,z;scanf("%d%d%d",&x,&y,&z);
        add(x,y,i);
        add(y,x,i);
        e[i]=z;
    }
    while(q--){
        int op;scanf("%d",&op);
        if(op==1){
            int x,y;scanf("%d%d",&x,&y);
            a[x]=y;
        }else if(op==2){
            int x,y;scanf("%d%d",&x,&y);
            e[x]=y;
        }else{
            cal();
        }
    }
}
signed main(){
    solve();
    return 0;
}

---