bzoj 2725 [Violet 6]故乡的梦

无向图，给定S和T，多次询问在删除某条边时两点间最短路

我的SPFA被卡了=.=。我们对所有不在最短路径的上的边 ，设它的起点为u 终点为v。那么对于这条边，包含该边的最优的S到T路径就应该是 S-S1-u-v-T1-T 其中S1,T1表示最短路径上的两个点。那么这样的一条最短路可以再S1-T1中某条边被删除时使用，我们用线段树维护这一过程 QWQ 数组开多了有点晕啊

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;
const int maxn=400000+10;
typedef long long LL;
struct tree{int l,r;LL s,lazy;}t[maxn<<2];
struct node{int to,next;LL len;int fr;bool o;}table[maxn<<1];
int tot=1,head[maxn];
void add(int a,int b,int c){table[++tot]=(node){b,head[a],c,a};head[a]=tot;}
int n,m,S,T,Q;
char s[1<<25];
inline long long read()
{
    static char *c=s;
    if(c==s) fread(c,1,1<<25,stdin);
    long long u=0;
    while(*c<48) ++c;
    while(*c>32) u=u*10+*c++-48;
    return u;
}
int pre[maxn],pos[maxn],cnt;
LL f[maxn][2];
bool v[maxn],use[maxn],iskeypoint[maxn];
typedef pair<LL,int>Pair;
void spfa()
{
    for(int i=1;i<=n;i++) f[i][0]=1e14;
    f[S][0]=0;
    priority_queue<Pair,vector<Pair>,greater<Pair> >Q;
    Q.push(make_pair(0,S));
    while(!Q.empty())
    {
        Pair top=Q.top();
        Q.pop();
        int x=top.second;
        if(v[x]) continue;
        for(int i=head[x];i;i=table[i].next)
        if(f[table[i].to][0]>f[x][0]+table[i].len&&!v[table[i].to])
        {
            pre[table[i].to]=i;
            f[table[i].to][0]=f[x][0]+table[i].len;
            Q.push(make_pair(f[table[i].to][0],table[i].to));
        }
    }
}
int ans[maxn][2],num[maxn];
void QWQ(int k)
{
    memset(v,0,sizeof v);
    for(int i=1;i<=n;i++) f[i][k]=1e14;
    priority_queue<Pair,vector<Pair>,greater<Pair> >Q;
    if(k==0) f[S][0]=0,Q.push(make_pair(0,S));
    else f[T][1]=0,Q.push(make_pair(0,T));
    while(!Q.empty())
    {
        Pair top=Q.top();
        Q.pop();
        int x=top.second;
        if(v[x]) continue;
        for(int i=head[x];i;i=table[i].next)
        if(f[table[i].to][k]>f[x][k]+table[i].len&&!v[table[i].to])
        {
            if(iskeypoint[table[i].to]&&iskeypoint[x]) ans[table[i].to][k]=i;
            else ans[table[i].to][k]=ans[x][k];
            f[table[i].to][k]=f[x][k]+table[i].len;
            Q.push(make_pair(f[table[i].to][k],table[i].to));
        }
    }
}
void maketree(int x,int l,int r)
{
    t[x].l=l; t[x].r=r;
    t[x].lazy=t[x].s=1e14;
    if(l==r) return ;
    int mid=l+r>>1;
    maketree(x<<1,l,mid);
    maketree(x<<1|1,mid+1,r);
    t[x].s=min(t[x<<1].s,t[x<<1|1].s);
}
void change(int x,int l,int r,LL tar)
{
    if(t[x].l==l&&t[x].r==r) 
    {
        t[x].lazy=min(t[x].lazy,tar);
        return ;
    }
    int mid=t[x].l+t[x].r>>1;
    int lc=x<<1,rc=x<<1|1;
    if(r<=mid) change(x<<1,l,r,tar);
    else if(l>mid) change(x<<1|1,l,r,tar);
    else change(x<<1,l,mid,tar),change(x<<1|1,mid+1,r,tar);
}
LL res;
void search(int x,int pos)
{
    res=min(res,t[x].lazy);
    if(t[x].l==t[x].r) return ;
    int mid=t[x].l+t[x].r>>1;
    int lc=x<<1,rc=x<<1|1;
    if(pos<=mid) return search(x<<1,pos);
    return search(x<<1|1,pos);
}
int main()
{
    scanf("%d%d",&n,&m);
    int x,y,c;
    for(int i=1;i<=m;i++)
    {
        x=read(); y=read(); c=read();
        add(x,y,c); add(y,x,c);
    }
    S=read(); T=read();
    spfa();
    int hh=T;
    while(hh)
    {
        if(pre[hh]) pos[++cnt]=pre[hh];
        use[pre[hh]]=1;
        iskeypoint[hh]=1;
        table[pre[hh]].o=1;
        hh=table[pre[hh]].fr;
    }
    for(int i=1;i<=cnt;i++) num[i]=pos[i];
    for(int i=1;i<=cnt;i++) pos[i]=num[cnt-i+1];
    memset(num,0,sizeof num);
    for(int i=1;i<=cnt;i++) num[pos[i]]=i;
    for(int i=1;i<=n;i++) ans[i][0]=-1,ans[i][1]=tot+1;
    QWQ(0); QWQ(1);
    maketree(1,1,cnt);
    for(int i=1;i<=tot;i++)
    if(!use[i])
    {
        int l=ans[table[i].fr][0],r=ans[table[i].to][1];
        int pos1,pos2;
        if(l==-1) pos1=1; else pos1=num[l]+1;
        if(r==tot+1) pos2=cnt; else pos2=num[r^1]-1;
        if(pos1<=pos2) change(1,pos1,pos2,table[i].len+f[table[i].fr][0]+f[table[i].to][1]);
    }
    Q=read();
    while(Q--)
    {
        x=read(); y=read();
        int pos;
        if((table[pre[x]].fr==y||table[pre[y]].fr==x)&&iskeypoint[x]&&iskeypoint[y]) 
        {
            if(table[pre[y]].fr==x) pos=num[pre[y]];
            else pos=num[pre[x]];
            res=1e14;
            search(1,pos);
            if(res>=1e14) puts("Infinity");
            else printf("%lld\n",res);
        }
        else printf("%lld\n",f[T][0]);
    }
    return 0;
}