1. 程式人生 > >[SDOI2010]魔法豬學院[k短路][A*]

[SDOI2010]魔法豬學院[k短路][A*]

題目描述

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。


K短路模板

//#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<cctype>
#include<queue>
#define N 1005
#define M 100005
using namespace std;
int first[N],next[M*2],to[M*2],w[M*2],tot;
int first2[N],next2[M*2],to2[M*2],w2[M*2],tot2;
int dis[N],vis[N],n,m,s,t,k;
struct Node{
	int pos,f,dis;
	bool operator < (Node a) const{
		return a.f+a.dis<f+dis;
	}
};
priority_queue<Node> q; 
int read(){
	int cnt=0;char ch=0;
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))cnt=cnt*10+(ch-'0'),ch=getchar();
	return cnt;
}
void add(int x,int y,int z){
	next[++tot]=first[x],first[x]=tot,to[tot]=y,w[tot]=z;
	next2[++tot2]=first2[y],first2[y]=tot2,to2[tot2]=x,w2[tot2]=z;
}
void spfa(){
	queue<int> q1;
	q1.push(t);
	memset(dis,0x3fffffff,sizeof(dis));
	dis[t]=0;
	while(!q1.empty()){
		int x=q1.front();
		q1.pop(),vis[x]=0;
		for(int i=first2[x];i;i=next2[i]){
			int T=to2[i];
			if(dis[T]>dis[x]+w2[i]){
				dis[T]=dis[x]+w2[i];
				if(!vis[T]) q1.push(T),vis[T]=1;
			}
		}
	}
}
int astar(){
	if(dis[s]==0x3fffffff) return -1;
	int times[N];
	memset(times,0,sizeof(times));
	Node tmp,h;
	h.pos=s,h.f=0,h.dis=0;
	q.push(h);
	while(!q.empty()){
		Node x=q.top(); q.pop();
		times[x.pos]++;
		if(times[x.pos]==k&&x.pos==t) return x.dis;
		if(times[x.pos]>k) continue;
		for(int i=first[x.pos];i;i=next[i]){
			tmp.dis=x.dis+w[i];
			tmp.f=dis[to[i]];
			tmp.pos=to[i];
			q.push(tmp);
		}
	}
	return -1;
}
int main(){
	n=read(),m=read();
	for(int i=1;i<=m;i++){
		int x=read(),y=read(),z=read();
		add(x,y,z);
	}
	s=read(),t=read(),k=read();
	if(s==t) k++;
	spfa();
	cout<<astar();
	return 0;
}