bzoj1598[Usaco2008 Mar]牛跑步 K短路模板 spfa+dij

29 篇文章 0 订阅

K短路的裸题。。
dij:

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<vector>
#include<ctime>
#include<cstdlib>
#include<queue> 
#include<ext/pb_ds/priority_queue.hpp>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define pa pair<int,int>
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define inf 2000000000
using namespace std;
using namespace __gnu_pbds;
const int N=1e5+5;
typedef __gnu_pbds::priority_queue<pa,greater<pa> > heap;
int n,m,K,cnt,cnt2;
int ans[N],d[N],g[N],tim[N];
int last[N],last2[N];
heap q;
heap::point_iterator id[N];
struct edge
{
    int to,next,v;
}e[N],ed[N];
inline void insert(int x,int y,int z)
{
    e[++cnt].to=y;e[cnt].next=last[x];e[cnt].v=z;last[x]=cnt;
    ed[++cnt2].to=x;ed[cnt2].next=last2[y];last2[y]=cnt2;ed[cnt2].v=z;
}
inline void dij()
{
    fo(i,1,n)d[i]=inf;
    d[1]=0;
    id[1]=q.push(make_pair(0,1));
    while(!q.empty())
    {
        int now=q.top().second;q.pop();
        for(int i=last2[now];i;i=ed[i].next)
        if (ed[i].v+d[now]<d[ed[i].to])
        {
            d[ed[i].to]=d[now]+ed[i].v;
            if (id[ed[i].to]!=0)
            q.modify(id[ed[i].to],make_pair(d[ed[i].to],ed[i].to));
            else id[ed[i].to]=q.push(make_pair(d[ed[i].to],ed[i].to));
        }
    }
}
inline void Astar()
{
    if (d[n]==inf)return;
    q.push(make_pair(d[n],n));
    while (!q.empty())
    {
        int now=q.top().second;
        int dis=q.top().first;
        q.pop();
        tim[now]++;
        if (now==1)ans[tim[now]]=dis;
        if (tim[now]<=K)
        for(int i=last[now];i;i=e[i].next)
        q.push(make_pair(dis-d[now]+d[e[i].to]+e[i].v,e[i].to)); 
    }
}
int main()
{
    scanf("%d%d%d",&n,&m,&K);
    fo(i,1,m)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        insert(x,y,z);
    }
    dij();
    Astar();
    fo(i,1,K)
    if (ans[i])printf("%d\n",ans[i]);
    else puts("-1");
    return 0;
}

spfa(来自zyf2000。。稍后我自己打一份。。)
呼呼。。终于打完了。。
cnt表示的是到这个点的路径数量,估价函数的话自己看看相关教程吧。
具体做法基本上就是先跑一边spfa,求出Astar中的f,然后求h和g的话Astar就好了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=2e5+5;
int n,m;
typedef long long ll;
ll dis[N],ans[N],val[N],val1[N];
int cnt[N];
int tot;
int s,t,k;
int head[N],next[N],go[N];
int tot1,head1[N],next1[N],go1[N];
struct node
{
    int x;
    ll g,h;
    bool operator < (const node &a) const
    {
        return g+h>a.g+a.h;
    }
};
inline void add(int x,int y,ll z)
{
    go[++tot]=y;next[tot]=head[x];val[tot]=z;head[x]=tot;
    go1[++tot1]=x;next1[tot1]=head1[y];val1[tot1]=z;head1[y]=tot1;
}
bool vis[N];
int q[N];
inline void spfa()
{
    memset(dis,127,sizeof(dis));
    dis[t]=0;
    vis[t]=1;
    q[1]=t;
    int t=0,w=1;
    while (t<w)
    {
        int x=q[++t];
        for(int i=head1[x];i;i=next1[i])
        {
            int v=go1[i];
            if (dis[v]>dis[x]+val1[i])
            {
                dis[v]=dis[x]+val1[i];
                if (!vis[v])vis[v]=1,q[++w]=v;
            }
        }
        vis[x]=0;
    }
}
inline void Astar()
{
    priority_queue <node> q;
    q.push((node){s,0,dis[s]});
    int size=0;
    while (!q.empty())
    {
        node x=q.top();
        q.pop();
        ++cnt[x.x];
        if (cnt[x.x]>k)continue;
        if (x.x==t)ans[++size]=x.g;
        if (cnt[t]==k)return;
        for(int i=head[x.x];i;i=next[i])
        q.push((node){go[i],x.g+val[i],dis[go[i]]});
    }
} 
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    fo(i,1,m)
    {
        int x,y;
        ll z;
        scanf("%d%d%lld",&x,&y,&z);
        if (x<y)swap(x,y);
        add(x,y,z);
    }
    s=n,t=1;
    spfa();
    memset(ans,-1,sizeof(ans));
    Astar();
    fo(i,1,k)printf("%lld\n",ans[i]);
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值