bzoj 1975: [Sdoi2010]魔法猪学院 k短路+A*

题意

给出n个点m条有向边，求最多有多少条1到n的不同路径使得每条路劲的权值加起来不超过c。
n<=5000,m<=200000

分析

很显然题目是要我们找一个最大的k满足1到n前k短的路径和不超过c。
那么我们就要来求k短路。

k短路

用spfa来实现，每次在优先队列里面取一个估价函数f(i)=g(i)+h(i)的点出来去更新其他的点，若一个点第k次出队则这次必然是第k短路。g(i)表示的是1到i当前的距离；h(i)表示的是i到n的最短距离，可以先预处理出来。

那么我们就每次找到一条k短路然后判断是否能更新就好了。

这题卡空间啊~~据说还不能用stl自带优先队列

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
#define N 5005
#define M 200005
#define inf 0x3f3f3f3f
using namespace std;

struct node
{
    int x;
    double d,f;
    bool operator < (const node &a) const
    {
        return f<a.f;
    }
}a[M*13];

struct edge{int to,next;double w;}e[M*2];

int n,m,ans,last[N],last1[N],vis[N],cnt,a1;
double c,h[N];
queue <int> q;

void addedge(int u,int v,double w)
{
    e[++cnt].to=v;e[cnt].w=w;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].w=w;e[cnt].next=last1[v];last1[v]=cnt;
}

void spfa()
{
    for (int i=1;i<=n;i++)
        h[i]=inf;
    h[n]=0;
    vis[n]=1;
    q.push(n);
    while (!q.empty())
    {
        int u=q.front();
        q.pop();
        for (int i=last1[u];i;i=e[i].next)
            if (h[u]+e[i].w<h[e[i].to])
            {
                h[e[i].to]=h[u]+e[i].w;
                if (!vis[e[i].to])
                {
                    vis[e[i].to]=1;
                    q.push(e[i].to);
                }
            }
        vis[u]=0;
    }
}

void push(node u)
{
    a[++a1]=u;
    int x=a1;
    while (x>1&&a[x]<a[x/2])
    {
        swap(a[x],a[x/2]);
        x/=2;
    }
}

node pop()
{
    node u=a[1];
    a[1]=a[a1];
    a1--;
    int x=1;
    while (a[x*2]<a[x]&&x*2<=a1||a[x*2+1]<a[x]&&x*2+1<=a1)
        if (a[x*2]<a[x*2+1]||x*2+1>a1)
        {
            swap(a[x],a[x*2]);
            x*=2;
        }else
        {
            swap(a[x],a[x*2+1]);
            x=x*2+1;
        }
    return u;
}

void Astar()
{
    node u;
    u.x=1;u.d=0;u.f=h[1];
    push(u);
    while (1)
    {
        node u=pop();
        for (int i=last[u.x];i;i=e[i].next)
        {
            node w;
            w.x=e[i].to;w.d=u.d+e[i].w;w.f=h[e[i].to]+w.d;
            push(w);
        }
        if (u.x==n)
        {
            c-=u.d;
            if (c<0) return;
            ans++;
        }
    }
}

int main()
{
    scanf("%d%d%lf",&n,&m,&c);
    for (int i=1;i<=m;i++)
    {
        int x,y;
        double z;
        scanf("%d%d%lf",&x,&y,&z);
        addedge(x,y,z);
    }
    spfa();
    Astar();
    printf("%d\n",ans);
    return 0;
}