POJ 3463 Sightseeing

最短路+次短路(Dijkstra+priority_queue)


题意是要求你找出最短路的条数+与最短路只差1的次短路的条数。


哭开始只会算最短路的条数,和次短路的长度。真是给次短路条数跪了。ORZ。其他人都用Dijkstra。我想试试SPFA。

然后大神说要把这个最短,次短都拿去push。而且要用最短来。priority_queue。优先队列。妈蛋,这不是优先队列优化的Dijkstra么。

改得无比忧伤。反正开始改来改去连样例都过不了。后来想着 如果最短可以更新,原来的最短就变成次短了。

思路来了,刷刷就写完了。WA……ORZ。。。最后对比大神的才知道。我push的只有节点和长度。需要把标记也进去。

OOOOOOOOORZ。


#include<cstdio>
#include<cstring>
#include<string>
#include<queue>
#include<algorithm>
#include<map>
#include<stack>
#include<iostream>
#include<list>
#include<set>
#include<cmath>
#define INF 0x7fffffff
#define eps 1e-6
#define LL long long
using namespace std;
int n,m;
struct lx
{
    int v,len;
};
vector <lx> g[1001];

struct node
{
    bool flag;
    int u;
    LL len;
    node(LL llen,int uu,bool fflag):len(llen),u(uu),flag(fflag){}

};
bool operator <(node a,node b)
{
    return a.len>b.len;
}
void Dijkstra(int start,int thend)
{
    bool vis[1001][2];
    int cot[1001][2];
    LL dis[1001][2];
    for(int i=1; i<=n; i++)
    {
        for(int j=0;j<2;j++)
        dis[i][j]=INF,cot[i][j]=0,vis[i][j]=0;;
    }
    priority_queue<node>q;
    q.push(node(0,start,0));
    dis[start][0]=0;
    cot[start][0]=cot[start][1]=1;

    while(!q.empty())
    {
        node now=q.top();q.pop();
        int u=now.u;
//        printf("%lld %d %d==\n",now.len,now.u,now.flag);
        if(vis[u][now.flag])continue;
        vis[u][now.flag]=1;

        for(int j=0;j<g[u].size();j++)
        {
            LL tmp=g[u][j].len+now.len;
            int v=g[u][j].v;

            if(dis[v][0]>tmp)
            {
                dis[v][1]=dis[v][0];
                cot[v][1]=cot[v][0];
                dis[v][0]=tmp;
                cot[v][0]=cot[u][now.flag];
                q.push(node(tmp,v,0));
                q.push(node(dis[v][1],v,1));

            }
            else if(dis[v][0]==tmp)
                cot[v][0]+=cot[u][now.flag];
            else if(dis[v][1]>tmp)
            {
                dis[v][1]=tmp;
                cot[v][1]=cot[u][now.flag];
                q.push(node(tmp,v,1));
            }
            else if(dis[v][1]==tmp)
                cot[v][1]+=cot[u][now.flag];
        }
    }
//    for(int i=1;i<=n;i++)
//        printf("%d :%lld-%lld\n",i,dis[i][0],dis[i][1]);


    if(dis[thend][0]+1!=dis[thend][1])
        printf("%d\n",cot[thend][0]);
    else
        printf("%d\n",cot[thend][0]+cot[thend][1]);
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1; i<=n; i++)
            g[i].clear();
        int u,v,len;
        lx now;
        while(m--)
        {
            scanf("%d%d%d",&u,&v,&len);
            now.v=v,now.len=len;
            g[u].push_back(now);
        }
        int start,thend;
        scanf("%d%d",&start,&thend);
        Dijkstra(start,thend);
    }
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值