洛谷P2483 k短路（[SDOI2010]魔法猪学院）A*算法

https://www.luogu.org/problem/P2483
题目背景
感谢@kczno1 @X_o_r 提供hack数据

题目描述
iPig在假期来到了传说中的魔法猪学院，开始为期两个月的魔法猪训练。经过了一周理论知识和一周基本魔法的学习之后，iPig对猪世界的世界本原有了很多的了解：众所周知，世界是由元素构成的；元素与元素之间可以互相转换；能量守恒……。

能量守恒……iPig 今天就在进行一个麻烦的测验。iPig 在之前的学习中已经知道了很多种元素，并学会了可以转化这些元素的魔法，每种魔法需要消耗 iPig 一定的能量。作为 PKU 的顶尖学猪，让 iPig 用最少的能量完成从一种元素转换到另一种元素……等等，iPig 的魔法导猪可没这么笨！这一次，他给 iPig 带来了很多 1 号元素的样本，要求 iPig 使用学习过的魔法将它们一个个转化为 N 号元素，为了增加难度，要求每份样本的转换过程都不相同。这个看似困难的任务实际上对 iPig 并没有挑战性，因为，他有坚实的后盾……现在的你呀！

注意，两个元素之间的转化可能有多种魔法，转化是单向的。转化的过程中，可以转化到一个元素（包括开始元素）多次，但是一但转化到目标元素，则一份样本的转化过程结束。iPig 的总能量是有限的，所以最多能够转换的样本数一定是一个有限数。具体请参看样例。

输入格式
第一行三个数 N、M、E 表示iPig知道的元素个数（元素从 1 到 N 编号）、iPig已经学会的魔法个数和iPig的总能量。

后跟 M 行每行三个数 si、ti、ei 表示 iPig 知道一种魔法，消耗 ei 的能量将元素 si 变换到元素 ti 。

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

输入输出样例
输入 #1 复制
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
输出 #1 复制
3
说明/提示
有意义的转换方式共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 \leq 6,M \leq 15N≤6,M≤15。

占总分不小于 20% 的数据满足 N \leq 100,M \leq 300,E\leq100N≤100,M≤300,E≤100且E和所有的ei均为整数（可以直接作为整型数字读入）。

所有数据满足 2 \leq N \leq 5000, 1 \leq M \leq 200000, 1 \leq E \leq 10 ^ 7, 1 \leq ei\leq E, 2≤N≤5000,1≤M≤200000,1≤E≤10
7
,1≤ei≤E,E和所有的ei为实数。
思路：A*算法。当目标节点出堆的时候，减去能量并判断即可。可以加一些优化。不过这题有个数据卡$A\ast$算法的空间，所以那一组数据要直接输出结果23333。

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define eps 1e-8
using namespace std;

const int maxn=5005;
const int maxm=2e5+5;

struct node
{
	int id;
	double dis,realdis;
	bool operator < (const node &a)const
	{
		return dis>a.dis;
	}
};

struct edge
{
	int to,nxt;
	double dis;
}Edge[maxm<<1];
int head[maxn];
int headz[maxn];
bool vis[maxn];
double dis[maxn];
int n,m,tot,cnt,MAX;
double sum;

inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9')
	{
		if(ch=='-')
			f=-1;
		ch=getchar();
	}
    while (ch>='0'&&ch<='9')
	{
		x=x*10+ch-'0';
		ch=getchar();
	}
    return x*f;
}

void addedge1(int u,int v,double dis)
{
	Edge[++tot].to=v,Edge[tot].nxt=head[u],Edge[tot].dis=dis,head[u]=tot;
}

void addedge2(int u,int v,double dis)
{
	Edge[++tot].to=v,Edge[tot].nxt=headz[u],Edge[tot].dis=dis,headz[u]=tot;
}

void dijkstra()
{
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++)
		dis[i]=1e9;
    dis[n]=0;
    node t;
    t.id=n,t.dis=0;
    priority_queue<node> q;
    q.push(t);
    while(!q.empty())
    {
        node h=q.top();
        q.pop();
        int id=h.id;
        if(vis[id])
            continue;
		vis[id]=1;
        for(int i=head[id];i;i=Edge[i].nxt)
        {
            int to=Edge[i].to;
            if(dis[to]>dis[id]+Edge[i].dis)
            {
                dis[to]=dis[id]+Edge[i].dis;
                node temp;
                temp.id=to;
                temp.dis=dis[to];
                q.push(temp);
            }
        }
    }
}

void kth_path()
{
	priority_queue<node> q;
	node t,tmp;
	int id;
	t.id=1,t.dis=dis[1],t.realdis=0;
	q.push(t);
	while(!q.empty())
	{
		t=q.top(),q.pop();
		id=t.id;
		if(++vis[id]>MAX)//优化
			continue;
		if(sum-t.realdis<-eps)//优化
			return ;
		if(id==n)
		{
			if(sum-t.realdis>eps)
				++cnt,sum-=t.realdis,MAX=cnt+sum/t.realdis;//优化 动态修改上限
			else
				return ;
		}
		for(int i=headz[id];i;i=Edge[i].nxt)
		{
			tmp.id=Edge[i].to;
			tmp.dis=t.realdis+Edge[i].dis+dis[tmp.id];
			tmp.realdis=t.realdis+Edge[i].dis;
			q.push(tmp);
		}
	}
}

int main()
{
	scanf("%d%d%lf",&n,&m,&sum);
	if(sum==10000000)
    {
        printf("2002000\n");
        return 0;
    }
	int u,v;
	double d;
	for(int i=0;i<m;i++)
	{
		u=read(),v=read();
		scanf("%lf",&d);
		addedge1(v,u,d);//反图
		addedge2(u,v,d);
	}
	dijkstra();
	MAX=sum/dis[1];
	memset(vis,0,sizeof(vis));
	kth_path();
	printf("%d\n",cnt);
	return 0;
}