HDU 6393

题意:n个点n条边的带权无向图(相当于一棵树多了一条边),两种操作:修改一条边的权值;求两点间的最短路径。

分析:将环上一条边取出来,其余n-1条边可以结合LCA解最短路。询问时,比较通过取出的边和仅通过树上的边的路径的大小,最小值就是两点的最短路径。

树状数组差分维护点到根节点的距离,根据dfs序来记录需要维护的范围。

不知道为什么欧拉序dfs序都能过
代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=2e5+5;
int n,m,cnt,tot,x,y,z,X,Y,V,ID,op,w,f[N][20],deep[N],l[N],r[N],p[N],v[N],head[N];
ll s[N<<1];
bool vis[N];
struct node{int nxt,id,to,v;}edge[N<<1];
inline ll min(ll x,ll y){return x<y?x:y;}
void addedge(int x,int y,int v,int id)
{edge[++cnt].nxt=head[x];edge[cnt].id=id;edge[cnt].to=y;edge[cnt].v=v;head[x]=cnt;}
void dfs(int x,int fa)
{
    vis[x]=1;
    l[x]=++tot;
    for(int i=head[x];i;i=edge[i].nxt)
    {
        int y=edge[i].to;
        if(y==fa) continue;
        if(vis[y])
        {
            X=x;Y=y;V=edge[i].v;
            ID=edge[i].id;
            continue;
        }
        deep[y]=deep[x]+1;
        f[y][0]=x;
        v[y]=edge[i].v;
        p[edge[i].id]=y;
        dfs(y,x);
    }
    r[x]=tot;
}
inline int lowbit(int x){return x&(-x);}
void update(int x,ll v){for(;x<=2*n;x+=lowbit(x)) s[x]+=v;}
ll query(int x)
{
    ll sum=0;
    for(;x;x-=lowbit(x)) sum+=s[x];
    return sum;
}
int lca(int x,int y)
{
    if(deep[x]<deep[y]) swap(x,y);
    for(int i=19;i>=0;i--) if(deep[f[x][i]]>=deep[y]) x=f[x][i];
    if(x==y) return x;
    for(int i=19;i>=0;i--) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
ll dis(int x,int y)
{
    int fa=lca(x,y);
    return query(l[x])+query(l[y])-2*query(l[fa]);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        addedge(x,y,z,i);
        addedge(y,x,z,i);
    }
    deep[1]=1;
    dfs(1,0);
    for(int i=1;(1<<i)<=n;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++)
    {
        update(l[i],v[i]);
        update(r[i]+1,-v[i]);
    }
    while(m--)
    {
        scanf("%d",&op);
        if(op==0)
        {
            scanf("%d%d",&x,&w);
            if(x!=ID)
            {
                x=p[x];
                update(l[x],w-v[x]);
                update(r[x]+1,v[x]-w);
                v[x]=w;
            }
            else V=w;
        }
        else
        {
            scanf("%d%d",&x,&y);
            ll ans=dis(x,y);
            ans=min(ans,dis(x,X)+dis(y,Y)+V);
            ans=min(ans,dis(x,Y)+dis(X,y)+V);
            printf("%lld\n",ans);
        }
    }
    return 0;
}
相关推荐
©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页