洛谷 P2149 [SDOI2009]Elaxia的路线（SPFA+记忆化搜索）

传送门

---

其实题目大意就是求两条最短路径的公共部分，那么怎么判断一条边(u,v,d)是否同时在两条最短路径内呢？
当

$$(dis_{x1,u}+d+dis_{v,y1}=dis_{x1,y1})∧(dis_{x2,u}+d+dis_{v,y2}=dis_{x2,y2})$$

时(u,v,d)同时在两条最短路径内。
这样我们可以对于x1,x2,y1,y2分别跑一次单源最短路然后对于满足条件的边重新构图，跑记忆化搜索求最长路。

Code:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<queue>
using namespace std;

const int INF=0x3f3f3f3f;

struct node{int x,y,c,next;}a[1000010],b[1000010];
int n,m,len(0),cnt(0),p[5],ans(0);
int first[1510],last[1510],f[1510],q[1510],d1[1510],d2[1510],d3[1510],d4[1510];
bool v[1510];

void ins(int x,int y,int c){a[++len]=(node){x,y,c,first[x]};first[x]=len;}
void eins(int x,int y,int c){b[++cnt]=(node){x,y,c,last[x]};last[x]=cnt;}

void spfa(int st,int *d)
{
    int head=1,tail=2;
    memset(v,false,sizeof(v));v[st]=true;
    d[st]=0;q[1]=st;
    while(head!=tail)
    {
        int x=q[head];
        for(int i=first[x];i;i=a[i].next)
        {
            int y=a[i].y;
            if(d[y]>d[x]+a[i].c)
            {
                d[y]=d[x]+a[i].c;
                if(!v[y])
                {
                    v[y]=true;
                    q[tail++]=y;
                    if(tail>n) tail=1;
                }
            }
        }
        v[x]=false;
        head++;
        if(head>n) head=1;
    }
}

void re_build()
{
    for(int i=1;i<=len;i++)
    {
        int x=a[i].x,y=a[i].y,c=a[i].c;
        if(d1[x]+c+d2[y]==d1[p[2]] && d3[x]+c+d4[y]==d3[p[4]] || d1[y]+c+d2[x]==d1[p[2]] && d3[x]+c+d4[y]==d3[p[4]])
            eins(x,a[i].y,a[i].c);
    }
}

int dfs(int x)
{
    if(f[x]) return f[x];
    for(int i=last[x];i;i=b[i].next)
        f[x]=max(f[x],dfs(b[i].y)+b[i].c);
    return f[x];
}

int main()
{
    memset(f,0,sizeof(f));
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++) d1[i]=d2[i]=d3[i]=d4[i]=INF,first[i]=last[i]=0;
    for(int i=1;i<=4;i++) scanf("%d",&p[i]);
    for(int i=1;i<=m;i++)
    {
        int x,y,c;
        scanf("%d %d %d",&x,&y,&c);
        ins(x,y,c);ins(y,x,c);
    }
    spfa(p[1],d1);
    spfa(p[2],d2);
    spfa(p[3],d3);
    spfa(p[4],d4);
    re_build();
    for(int i=1;i<=n;i++) if(f[i]==0) dfs(i);
    for(int i=1;i<=n;i++) ans=max(ans,f[i]);
    printf("%d",ans);
}