BZOJ_1975_[Sdoi2010]魔法猪学院_A*

BZOJ_1975_[Sdoi2010]魔法猪学院_A*

Description

iPig在假期来到了传说中的魔法猪学院，开始为期两个月的魔法猪训练。经过了一周理论知识和一周基本魔法的学习之后，iPig对猪世界的世界本原有了很多的了解：众所周知，世界是由元素构成的；元素与元素之间可以互相转换；能量守恒……。 能量守恒……iPig 今天就在进行一个麻烦的测验。iPig 在之前的学习中已经知道了很多种元素，并学会了可以转化这些元素的魔法，每种魔法需要消耗 iPig 一定的能量。作为 PKU 的顶尖学猪，让 iPig 用最少的能量完成从一种元素转换到另一种元素……等等，iPig 的魔法导猪可没这么笨！这一次，他给 iPig 带来了很多 1 号元素的样本，要求 iPig 使用学习过的魔法将它们一个个转化为 N 号元素，为了增加难度，要求每份样本的转换过程都不相同。这个看似困难的任务实际上对 iPig 并没有挑战性，因为，他有坚实的后盾……现在的你呀！ 注意，两个元素之间的转化可能有多种魔法，转化是单向的。转化的过程中，可以转化到一个元素（包括开始元素）多次，但是一但转化到目标元素，则一份样本的转化过程结束。iPig 的总能量是有限的，所以最多能够转换的样本数一定是一个有限数。具体请参看样例。

Input

第一行三个数 N、M、E 表示iPig知道的元素个数（元素从 1 到 N 编号）、iPig已经学会的魔法个数和iPig的总能量。 后跟 M 行每行三个数 si、ti、ei 表示 iPig 知道一种魔法，消耗 ei 的能量将元素 si 变换到元素 ti 。

Output

一行一个数，表示最多可以完成的方式数。输入数据保证至少可以完成一种方式。

Sample Input

4 6 14.9
1 2 1.5
2 1 1.5
1 3 3
2 3 1.5
3 4 1.5
1 4 1.5

Sample Output

3

---

 

题意很明确，问最多能跑几次，其实就是K短路，每次判断长度和生育能量的大小即可。

K短路这里使用A*(可持久化可并堆做法挖坑代填)

估价函数设置为当前走过的距离+最短到终点的距离。

最短到终点的距离用dij预处理出来。

 

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <stdlib.h>
#include <ext/pb_ds/priority_queue.hpp>
using namespace std;
using namespace __gnu_pbds;
inline int rd() {
    register int x=0;
    register char s=getchar();
    while(s<'0'||s>'9') s=getchar();
    while(s>='0'&&s<='9') x=(x<<3)+(x<<1)+s-'0',s=getchar();
    return x;
}
typedef double f2;
#define N 5050
#define M 200050
int head[N],to[M],nxt[M],cnt,n,m,xx[M],yy[M],vis[N],tot[N],ans;
f2 val[M],zz[M],dis[N],lft;
__attribute__((optimize("-O2")))struct node {
    f2 v;
    int x;
    node() {}
    node(f2 v_,int x_) :
        v(v_),x(x_) {}
    inline bool operator < (const node &u) const {
        return v+dis[x]>u.v+dis[u.x];
    }
};
__gnu_pbds::priority_queue<pair<f2,int> >q1;
__gnu_pbds::priority_queue<node>q2;
__attribute__((optimize("-O2")))inline void add(int u,int v,f2 w) {
    to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt; val[cnt]=w;
}
__attribute__((optimize("-O2")))void dij() {
    memset(dis,0x7f,sizeof(dis));
    dis[n]=0; q1.push(make_pair(0,n));
    while(!q1.empty()) {
        int x=q1.top().second,i; q1.pop();
        if(vis[x]) continue;
        vis[x]=1;
        for(i=head[x];i;i=nxt[i]) {
            if(dis[to[i]]>dis[x]+val[i]) {
                dis[to[i]]=dis[x]+val[i];
                q1.push(make_pair(-dis[to[i]],to[i]));
            }
        }
    }
}
__attribute__((optimize("-O2")))void _A_stAr_SapPh1r3_() {
    f2 inf=lft/dis[1];
    q2.push(node(0,1));
    while(!q2.empty()) {
        node t=q2.top(); q2.pop();
        int x=t.x;
        f2 v=t.v;
        // printf("%f\n",lft);
        int i;
        if(x==n) {
            if(lft>=v) {
                lft-=v; ans++;
            }else {
                printf("%d\n",ans); exit(0);
            }
        }
        for(i=head[x];i;i=nxt[i]) {
            q2.push(node(v+val[i],to[i]));
        }
    }
}
__attribute__((optimize("-O2")))int main() {
    n=rd(); m=rd(); scanf("%lf",&lft);
    int i;
    for(i=1;i<=m;i++) {
        xx[i]=rd(); yy[i]=rd(); scanf("%lf",&zz[i]);
        add(yy[i],xx[i],zz[i]);
    }
    dij();
    //lft/=dis[1];
    memset(head,0,sizeof(head)); cnt=0;
    for(i=1;i<=m;i++) {
        add(xx[i],yy[i],zz[i]);
    }
    _A_stAr_SapPh1r3_();
    printf("%d\n",ans);
}