POJ-2249-Remmarguts' Date

这个题要求求出第k短路，开始一直没啥好的思路， 后来才知道这个题要用A*做，虽然看了资料也没太明白，后来参考了别人的博客才做的。

地址：http://blog.sina.com.cn/s/blog_691ce2b701017fe3.html

我用的Spfa做的

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
using namespace std;
const int maxn=1e6+100;
const int maxm=1010;
struct node
{
    int to;
    int disa;
    int disb;
    bool operator <(const node &a)const
    {
	return  disa+disb>a.disa+a.disb;
    }
};
int e,n,m,head[maxm],tail[maxm],nxt[2*maxn],cost[2*maxn],pnt[2*maxn],dist[maxm],cnt[maxm];
int st,ed,k;
bool vis[maxm];
queue<int> q;
priority_queue<node> priq;
void AddEdge(int u,int v,int c)
{
    pnt[e]=v;nxt[e]=head[u];cost[e]=c;head[u]=e++;
    pnt[e]=u;nxt[e]=tail[v];cost[e]=c;tail[v]=e++;
}
void Spfa(int st)
{
    memset(dist,0x7f,sizeof(dist));
    dist[st]=0;
    memset(vis,0,sizeof(vis));
    q.push(st);
    while(!q.empty())
    {
	int u=q.front();
	vis[u]=0;
	q.pop();
	for(int i=tail[u];i!=-1;i=nxt[i])
	{
	    int v=pnt[i];
	    if(dist[v]>dist[u]+cost[i])
	    {
		dist[v]=dist[u]+cost[i];
		if(!vis[v])
		{
		    q.push(v);
		    vis[v]=1;
		}
	    }
	}
    }
}
int AStra(int k)
{
    while(!priq.empty())
	priq.pop();
    memset(cnt,0,sizeof(cnt));
    node ita;
    ita.to=st;
    ita.disa=0;
    ita.disb=dist[st];
    priq.push(ita);
    while(!priq.empty())
    {
	node u=priq.top();
	priq.pop();
	if(++cnt[u.to]>k)
	    continue;
	if(cnt[ed]==k)
	    return u.disa;
	for(int i=head[u.to];i!=-1;i=nxt[i])
	{
	    int v=pnt[i];
	    ita.to=v;
	    ita.disa=u.disa+cost[i];
	    ita.disb=dist[v];
	    priq.push(ita);
	}
    }
    return -1;
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
	memset(head,-1,sizeof(head));
	memset(tail,-1,sizeof(tail));
	e=0;
	for(int i=0;i<m;i++)
	{
	    int u,v,c;
	    scanf("%d%d%d",&u,&v,&c);
	    AddEdge(u,v,c);
	}
	scanf("%d%d%d",&st,&ed,&k);
	if(st==ed)
	    k++;
	Spfa(ed);
	printf("%d\n",AStra(k));
    }
    return 0;
}