bzoj2725 [Violet 6]故乡的梦（最短路树+线段树）

每次询问删去一条边之后的S-T最短路。
我们首先搞出最短路树。然后对于每一条在S-T最短路上的边都处理出删掉他之后的答案。类似bzoj4400

然后对于询问O(1)回答即可。

此题需要注意一些特判：比如S==T S,T一开始就不联通。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <utility>
using namespace std;
#define ll long long
#define N 200010
#define pa pair<ll,int>
ll const inf=1LL<<60;
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,h[N],num=1,mm,S,T,path[N],a[N],tot=0,dep[N];
ll dis[N][2],ans[N];
bool inq[N],mark[N];
struct edge{
    int to,next,val;
}data[N<<1];
struct node{
    ll mn;
}tr[N<<2];
inline void Dijkstra1(){
    priority_queue<pa,vector<pa>,greater<pa> >q;
    for(int i=1;i<=n;++i) dis[i][0]=dis[i][1]=inf;
    q.push(make_pair(0,S));dis[S][0]=0;
    while(!q.empty()){
        int x=q.top().second;q.pop();
        if(inq[x]) continue;inq[x]=1;
        for(int i=h[x];i;i=data[i].next){
            int y=data[i].to;
            if(dis[x][0]+data[i].val<dis[y][0]){
                dis[y][0]=dis[x][0]+data[i].val;path[y]=i;
                q.push(make_pair(dis[y][0],y));
            }
        }
    }
}
inline void Dijkstra2(){
    priority_queue<pa,vector<pa>,greater<pa> >q;memset(inq,0,sizeof(inq));
    q.push(make_pair(0,T));dis[T][1]=0;
    while(!q.empty()){
        int x=q.top().second;q.pop();
        if(inq[x]) continue;inq[x]=1;
        for(int i=h[x];i;i=data[i].next){
            int y=data[i].to;
            if(dis[x][1]+data[i].val<dis[y][1]){
                dis[y][1]=dis[x][1]+data[i].val;
                q.push(make_pair(dis[y][1],y));
            }
        }
    }
}
inline void gao(){
    int x=T;a[++tot]=T;
    while(path[x]){mark[path[x]>>1]=1;x=data[path[x]^1].to;a[++tot]=x;}
    for(int i=1,j=tot;i<j;++i,--j) swap(a[i],a[j]);
    for(int i=1;i<=tot;++i) dep[a[i]]=i;
    for(int i=1;i<=n;++i){
        if(dep[i]) continue;
        for(x=i;!dep[x];x=data[path[x]^1].to);
        for(int xx=i;!dep[xx];xx=data[path[xx]^1].to) dep[xx]=dep[x];
    }
}
inline void build(int p,int l,int r){
    tr[p].mn=inf;if(l==r) return;int mid=l+r>>1;
    build(p<<1,l,mid);build(p<<1|1,mid+1,r);
}
inline void change(int p,int l,int r,int x,int y,ll val){
    if(x<=l&&r<=y){tr[p].mn=min(tr[p].mn,val);return;}
    int mid=l+r>>1;
    if(x<=mid) change(p<<1,l,mid,x,y,val);
    if(y>mid) change(p<<1|1,mid+1,r,x,y,val);
}
inline void pushdown(int p){
    tr[p<<1].mn=min(tr[p<<1].mn,tr[p].mn);
    tr[p<<1|1].mn=min(tr[p<<1|1].mn,tr[p].mn);
}
inline void ask(int p,int l,int r){
    if(l==r){ans[l]=tr[p].mn;return;}
    int mid=l+r>>1;pushdown(p);
    ask(p<<1,l,mid);ask(p<<1|1,mid+1,r);
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();
    for(int i=1;i<=m;++i){
        int x=read(),y=read(),val=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=val;
        data[++num].to=x;data[num].next=h[y];h[y]=num;data[num].val=val;
    }S=read();T=read();
    if(S==T){
        m=read();while(m--) puts("0");return 0;
    }Dijkstra1();
    if(dis[T][0]==inf){
        m=read();while(m--) puts("Infinity");return 0;
    }Dijkstra2();gao();build(1,1,tot-1);
    for(int i=1;i<=m;++i){
        if(mark[i]) continue;int x=data[i<<1].to,y=data[i<<1|1].to;
        if(dep[x]==dep[y]) continue;if(dep[x]>dep[y]) swap(x,y);
        change(1,1,tot-1,dep[x],dep[y]-1,dis[x][0]+data[i<<1].val+dis[y][1]);
    }ask(1,1,tot-1);m=read();
    while(m--){
        int x=read(),y=read();if(dep[x]>dep[y]) swap(x,y);
        if(data[path[y]^1].to!=x||dep[x]+1!=dep[y]){printf("%lld\n",dis[T][0]);continue;}
        if(ans[dep[x]]==inf) puts("Infinity");
        else printf("%lld\n",ans[dep[x]]);
    }return 0;
}