BZOJ 1975: [Sdoi2010]魔法猪学院 K短路 dijkstra A*

1975: [Sdoi2010]魔法猪学院

Time Limit: 10 Sec  Memory Limit: 64 MB
Submit: 2235  Solved: 696
[Submit][Status][Discuss]

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

HINT

样例解释
有意义的转换方式共4种：
1->4，消耗能量 1.5
1->2->1->4，消耗能量 4.5
1->3->4，消耗能量 4.5
1->2->3->4，消耗能量 4.5
显然最多只能完成其中的3种转换方式（选第一种方式，后三种方式仍选两个），即最多可以转换3份样本。
如果将 E=14.9 改为 E=15，则可以完成以上全部方式，答案变为 4。

数据规模
占总分不小于 10% 的数据满足 N <= 6，M<=15。
占总分不小于 20% 的数据满足 N <= 100，M<=300，E<=100且E和所有的ei均为整数（可以直接作为整型数字读入）。
所有数据满足 2 <= N <= 5000，1 <= M <= 200000，1<=E<=107，1<=ei<=E，E和所有的ei为实数。

这个题，就是K短路

听说会卡priority_queue吓得我赶快手写左偏树

但写到astar还是因为懒人特质弃疗了

but...谁说卡空间的...明明能过...

#include<ctime>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;
typedef double db;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}
const int M=200100;
const int N=5010;
int n,m,ecnt,ecnt2,last[N],last2[N],l[N],r[N],root,ans;
bool book[N];
db h,use=0,dis[N],f[N];
struct EDGE{int to,nt;db val;}e[M],e2[M];
inline void add(int u,int v,db val)
{e[++ecnt]=(EDGE){v,last[u],val};last[u]=ecnt;}
inline void add2(int u,int v,db val)
{e2[++ecnt2]=(EDGE){v,last2[u],val};last2[u]=ecnt2;}
int merger(int x,int y)
{
    if(!x)return y;
    if(!y)return x;
    if(f[x]>f[y])swap(x,y);
    r[x]=merger(r[x],y);
    if(dis[r[x]]>dis[l[x]])swap(r[x],l[x]);
    dis[x]=dis[r[x]]+1;
    return x;
}
void dijkstra()
{
    memset(f,0X7f,sizeof(f));
    book[n]=1;f[n]=0;root=merger(root,n);
    while(root)
    {
        int u=root;root=merger(r[root],l[root]);
        for(int i=last2[u];i;i=e2[i].nt)
        if(f[e2[i].to]>f[u]+e2[i].val)
        {
            f[e2[i].to]=f[u]+e2[i].val;
            if(!book[e2[i].to])book[e2[i].to]=1,root=merger(e2[i].to,root);
        }
    }
}
struct P{db dis;int pos;};
bool operator <(P x,P y){return x.dis>y.dis;}
priority_queue<P>Q;
void astar()
{
    for(int i=1;i<=n;i++)dis[i]=f[i];
    P u,v;u.dis=dis[1];u.pos=1;Q.push(u);
    while(!Q.empty())
    {
        u=Q.top();Q.pop();
        int x=u.pos;
        if(x==n)
        {
            use+=u.dis;
            if(use>h)return ;
            ans++;continue;
        }
        for(int i=last[x];i;i=e[i].nt)
        {
            v.pos=e[i].to;v.dis=u.dis-dis[x]+e[i].val+dis[e[i].to];
            Q.push(v);
        }
    }
}
int main()
{
    n=read();m=read();scanf("%lf",&h);int u,v;db val;
    for(int i=1;i<=m;i++)
    {
        u=read();v=read();scanf("%lf",&val);
        add(u,v,val);add2(v,u,val);
    }
    dijkstra();astar();
    printf("%d\n",ans);
}
/*
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

3
*/