关闭

poj2449---BFS+Astar

标签: struct
701人阅读 评论(0) 收藏 举报
分类:

寻找第k短路。看到这道题时,很快就想到了思路,但就是实现不了。看了discuss,可以先用dijkstra求出最短路作为估价函数,在astar来实现求出第k短路,而astar最惯常做法是用优先队列保存节点。

用TPoint来保存astar里的节点,其中to是指向的位置,dis代表当前节点中已经过的距离,all是到目标的总距离。

#include<queue>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn 1005
#define inf 1<<30
int sum1[maxn],sum2[maxn];
int dis[maxn],cnt[maxn];
bool vis[maxn];
int tot1,tot2,n,m,t;
int st,en,kth;
struct TNode
{
	int to,next,cost;
	bool operator<(const TNode &temp) const
	{
		return cost>temp.cost;
	}
};
struct TPoint
{
	int to,dis,all;
	bool operator<(const TPoint &temp) const
	{
		return all>temp.all;
	}
};
vector<TNode>ans1,ans2;
priority_queue<TNode>qu;
priority_queue<TPoint>ans;
void addedge(int u,int v,int w,bool flag)
{
	TNode tmp;
	tmp.to=u,tmp.cost=w;
	if(flag)
	{
		tmp.next=sum1[v],sum1[v]=tot1++;
		ans1.push_back(tmp);
	}
	else
	{
		tmp.next=sum2[v],sum2[v]=tot2++;
		ans2.push_back(tmp);
	}
}
void dijkstra()
{
	int k;
	fill(dis,dis+n+1,inf);
	memset(vis,0,sizeof(vis));
	while(!qu.empty()) qu.pop();
	TNode tmp1,tmp2;
	tmp1.to=en,tmp1.cost=0;
	qu.push(tmp1);
	dis[en]=0;
	while(!qu.empty())
	{
		tmp1=qu.top(),qu.pop();
		if(vis[tmp1.to]) continue;
		vis[tmp1.to]=1;
		for(k=sum1[tmp1.to];k!=-1;k=ans1[k].next)
		{
		    if(dis[ans1[k].to]>dis[tmp1.to]+ans1[k].cost)
		    {
		        dis[ans1[k].to]=dis[tmp1.to]+ans1[k].cost;
		        tmp2.to=ans1[k].to;
		        tmp2.cost=dis[tmp2.to];
		        qu.push(tmp2);
		    }
		}
	}
}
int astar()
{
    int i,j,k;
    TPoint tmp1,tmp2;
    while(!ans.empty()) ans.pop();
    memset(cnt,0,sizeof(cnt));
    tmp1.to=st,tmp1.dis=0,tmp1.all=dis[st];
    if(tmp1.all==inf) return -1;
    ans.push(tmp1);
    while(!ans.empty())
    {
        tmp1=ans.top(),ans.pop();
        cnt[tmp1.to]++;
        if(cnt[en]==kth) return tmp1.all;
        if(cnt[tmp1.to]>kth) continue;
        for(k=sum2[tmp1.to];k!=-1;k=ans2[k].next)
        {
            tmp2.to=ans2[k].to;
            tmp2.dis=tmp1.dis+ans2[k].cost;
            tmp2.all=tmp2.dis+dis[tmp2.to];
            ans.push(tmp2);
        }
    }
    return -1;
}
int main()
{
	int i,u,v,w;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		tot1=tot2=0;
		memset(sum1,-1,sizeof(sum1));
		memset(sum2,-1,sizeof(sum2));
		ans1.clear();
		ans2.clear();
		for(i=0;i<m;i++)
		{
			scanf("%d%d%d",&u,&v,&w);
			addedge(u,v,w,1);
			addedge(v,u,w,0);
		}
		scanf("%d%d%d",&st,&en,&kth);
		dijkstra();
		if(st==en) kth++;
		printf("%d\n",astar());
	}
	return 0;
}


0
0

猜你在找
【直播】计算机视觉原理及实战—屈教授
【套餐】深度学习入门视频课程—唐宇迪
【套餐】Hadoop生态系统零基础入门--侯勇蛟
【套餐】嵌入式Linux C编程基础--朱有鹏
【套餐】2017软考系统集成项目——任铄
【套餐】Android 5.x顶级视频课程——李宁
【直播】广义线性模型及其应用——李科
【直播】从0到1 区块链的概念到实践
【直播】机器学习之凸优化——马博士
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:32060次
    • 积分:730
    • 等级:
    • 排名:千里之外
    • 原创:44篇
    • 转载:1篇
    • 译文:0篇
    • 评论:1条
    文章分类
    文章存档
    最新评论
  • poj1656----二维线段树

    alpczy: 你这样开第一维线段树有什么意义?查询第一维变成了log(n),还不如直接在第一维用数组,访问还是o(...