题意: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;
}