ACM-ICPC 2018 沈阳赛区网络预赛 D. Made In Heaven （K短路算法模板）

判断第k短路的权值是否小于T

可以用A_star模板过

#include<bits/stdc++.h>
#define maxn 1005
#define maxe 100005
using namespace std;

struct State{
    int f,g,u;  // f=g+dis dis表示当前点到终点的最短路径，即之前的预处理,g表示到当前点的路径长度
    bool operator<(const State b)const{
        if(f==b.f)return g>b.g;
        return f>b.f;
    }
};

struct Edge{
    int v,w,next;
}edge[maxe],reedge[maxe];

int head[maxn],rehead[maxn];
int dis[maxn],vis[maxn];
int n,m,cot,s,t,k;

void init(){
    cot=0;    //cot代表边的id  每条边的id都不相同
    memset(head,-1,sizeof(head));
    memset(rehead,-1,sizeof(rehead));
}

void addedge(int u,int v,int w){
    edge[cot]={v,w,head[u]};//记录上一次u的id是多少 这样方便遍历head[u]初始值为-1
    head[u]=cot;//head[u]  给这个u标记上独一无二的id
    reedge[cot]={u,w,rehead[v]};
    rehead[v]=cot++;
}

void SPFA(){
    queue<int>q;
    memset(vis,0,sizeof(vis));
    memset(dis,-1,sizeof(dis));
    int u,v;
    q.push(t);
    vis[t]=true;//vis表示当前点是否在队列
    dis[t]=0;
    while(!q.empty()){
        u=q.front();
        q.pop();
        vis[u] = 0;//rehead[u] 是u最后一次出现的id  reedge[i].ne  代表第i次出现的边上一次出现的id
        for(int i=rehead[u];~i;i=reedge[i].next){  //~i取反 当i为-1时正好取反为0 退出for
            v=reedge[i].v;
            if(dis[v]>dis[u]+reedge[i].w || dis[v]==-1){
                dis[v]=dis[u]+reedge[i].w;
                if(!vis[v]){
                    q.push(v);
                    vis[v]=1;
                }
            }
        }
    }
}

int Astart(){
    if(s==t)k++;
    if(dis[s]==-1)return -1;
    int cnt=0;
    priority_queue<State>q; // 优先队列
    State a,b;
    a.g=0;
    a.u=s;
    a.f=a.g+dis[a.u];
    q.push(a);
    while(!q.empty()){
        b=q.top();
        q.pop();
        if(b.u==t){
            cnt++;
            if(cnt==k)return b.g;
        }
        for(int i=head[b.u];~i;i=edge[i].next){
            a.g=b.g+edge[i].w;
            a.u=edge[i].v;
            a.f=a.g+dis[a.u];
            q.push(a);
        }
    }
    return -1;
}

int main()
{
    int u,v,w,T;
    while(scanf("%d%d",&n,&m)==2){
        init();
        scanf("%d%d%d%d",&s,&t,&k, &T);  //起点 终点 第k条 时间T
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&u,&v,&w);
            addedge(u,v,w);
        }
        SPFA();

        if (Astart() <= T && Astart() != -1) //小于等于T 并且有路才可以
            printf("yareyaredawa\n");
        else printf("Whitesnake!\n");
    }
    return 0;
}

 

转载于:https://www.cnblogs.com/wangtao971115/p/10358302.html