【BZOJ1975】[Sdoi2010]魔法猪学院

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

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为实数。

题解：A*算法求k短路，先在反图上跑一边SPFA求最短路，记每个点到终点的最短距离为h[i]，然后再正着跑Dijkstra，记从1到每个点的已经走得距离为g[i]，将估价函数设定为g[i]+h[i]即可。

优化：当每个点入队k+1次时，不将其进入队列。但是k不是固定的，我们记每个点入队的次数为vis[i]，我们令k=vis[n]+E/g[n]即可。

#include <cstdio>

#include <cstring>

#include <iostream>

#include <queue>

#include <cmath>

using namespace std;

int n,m,cnt,K;

double E;

const int maxn=5010;

const int maxm=200010;

double h[maxn],val[maxm],vv[maxm];

int to[maxm],next[maxm],head[maxn],tt[maxm],nn[maxm],hh[maxn],inq[maxn],vis[maxn];

struct point

{

	int v;

	double g;

	point() {}

	point(int a,double b) {v=a,g=b;}

};

bool operator < (point a,point b)

{

	return h[a.v]+a.g>h[b.v]+b.g;

}

void add(int a,int b,double c)

{

	to[cnt]=b,val[cnt]=c,next[cnt]=head[a],head[a]=cnt;

	tt[cnt]=a,vv[cnt]=c,nn[cnt]=hh[b],hh[b]=cnt++;

}

queue<int> q;

priority_queue<point> pq;

void spfa()

{

	int i,u;

	for(i=1;i<n;i++)	h[i]=99999999.99;

	q.push(n);

	while(!q.empty())

	{

		u=q.front(),q.pop(),inq[u]=0;

		for(i=hh[u];i!=-1;i=nn[i])

		{

			if(h[tt[i]]>h[u]+vv[i])

			{

				h[tt[i]]=h[u]+vv[i];

				if(!inq[tt[i]])	inq[tt[i]]=1,q.push(tt[i]);

			}

		}

	}

}

void astar()

{

	pq.push(point(1,0)),K=floor(E/h[1]);

	point t;

	int i,u;

	while(!pq.empty())

	{

		t=pq.top(),pq.pop(),u=t.v;

		if(vis[u]>=K)	continue;

		if(u==n)

		{

			if(E<t.g)	return ;

			K=vis[n]+floor(E/(t.g)),E-=t.g;

		}

		vis[u]++;

		for(i=head[u];i!=-1;i=next[i])

		{

			if(t.g+val[i]+h[to[i]]<=E&&vis[to[i]]<K)	pq.push(point(to[i],t.g+val[i]));

		}

	}

}

int main()

{

	int i,a,b;

	double c;

	scanf("%d%d%lf",&n,&m,&E);

	memset(head,-1,sizeof(head)),memset(hh,-1,sizeof(hh));

	for(i=1;i<=m;i++)

	{

		scanf("%d%d%lf",&a,&b,&c);

		add(a,b,c);

	}

	spfa(),astar();

	printf("%d",vis[n]);

	return 0;

}

巴特西

【BZOJ1975】[Sdoi2010]魔法猪学院 A*