题目:给你一个n边n点的无向连通图,两个操作,操作一改变某个边的权值,操作二查询某两个点之间的最短路径长度。
思路:在环上拿出一条边来放一边,搞一棵树出来,因为两点间最短距离就是是两点各自到根的距离之和减去 2*两点的lca到根的距离。 修改操作就变为维护点到根的这条链上的权值,差分得到欧拉序,用树状数组维护。查询的时候直接查询距离,再加上拿出来的那条不用的边情况,取最小即可。
找了好久的错误,发现竟然是因为没去 freopen ???
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
int t,n,q,N;
struct node{
ll w;
int to,net,id;
}edge[maxn<<1];
int first[maxn],tot;
void addedge(int u,int v,ll w,int id)
{
edge[++tot].to=v;
edge[tot].w=w;
edge[tot].id=id;
edge[tot].net=first[u];
first[u]=tot;
}
int book[maxn],X,Y,kt;
ll c[maxn<<1],W,w[maxn];
int L[maxn],R[maxn],f[maxn][20],cnt,dep[maxn],p[maxn];
void init()
{
tot=cnt=0;
N=n+n;
for(int i=0;i<=n;i++)
{
w[i]=0;
first[i]=-1;
book[i]=0;
for(int j=0;j<20;j++)
f[i][j]=0;
}
for(int i=0;i<=N;i++)
c[i]=0;
}
void dfs(int u,int pre)
{
book[u]=1;
L[u]=++cnt;
for(int i=first[u];i!=-1;i=edge[i].net)
{
int v=edge[i].to;
if(v==pre) continue;
if(book[v])
{
X=u;Y=v;W=edge[i].w;
kt=edge[i].id;
continue;
}
dep[v]=dep[u]+1;
f[v][0]=u;
w[v]=edge[i].w;
p[edge[i].id]=v;
dfs(v,u);
}
R[u]=++cnt;
}
int lowbit(int x){return x&(-x);}
void update(int pos,ll val)
{
for(;pos<=N;pos+=lowbit(pos))
c[pos]+=val;
}
ll query(int pos)
{
ll sum=0;
for(;pos>0;pos-=lowbit(pos))
sum+=c[pos];
return sum;
}
int lca(int x,int y)
{
if(dep[x]<dep[y]) swap(x,y);
int d=dep[x]-dep[y];
for(int i=19;i>=0;i--)
if(d&(1<<i))
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 dist(int x,int y)
{
int fa=lca(x,y);
ll ans=query(L[x])+query(L[y])-2*query(L[fa]);
return ans;
}
int main()
{
//freopen("in.txt","r",stdin);
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&q);
init();
for(int i=1;i<=n;i++)
{
int x,y;ll z;
scanf("%d%d%I64d",&x,&y,&z);
addedge(x,y,z,i);
addedge(y,x,z,i);
}
dep[1]=1;
dfs(1,-1);
for(int j=1;j<20;j++)
for(int i=1;i<=n;i++)
f[i][j]=f[f[i][j-1]][j-1];
for(int i=1;i<=n;i++)
{
update(L[i],w[i]);
update(R[i]+1,-w[i]);
}
for(int i=1;i<=q;i++)
{
int op,x,y;ll z;
scanf("%d",&op);
if(op==0)
{
scanf("%d%I64d",&x,&z);
if(x!=kt)
{
x=p[x];
update(L[x],z-w[x]);
update(R[x]+1,w[x]-z);
w[x]=z;
}
else W=z;
}
else
{
scanf("%d%d",&x,&y);
ll ans=dist(x,y);
ans=min(ans,dist(x,X)+dist(y,Y)+W);
ans=min(ans,dist(x,Y)+dist(y,X)+W);
printf("%I64d\n",ans);
}
}
}
return 0;
}