ZOJ2923 Calculate Roads(SPFA上的dp)

 

算是学了图dp后的第一次应用吧。题目其实真的是非常不严谨，什么都没说，基本靠猜，而且严格来说数据应该会有爆int的，不过不管那么多啦，思路对了就好－ －0

#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<vector>
#include<cmath>
#include<queue>
#define ll long long
#define maxn 5000
#define maxm 1000000
#define inf 0x3f3f3f3f
using namespace std;


vector<int> G[maxn+50];

struct Node
{
	int v,k;
	Node(){}
	Node(int vi,int ki):v(vi),k(ki){}
};

int m,n,k;
int dis[maxn+50][55];
int num[maxn+50][55];
int vis[maxn+50][55];
int vtype[maxn];

int main()
{
	while(cin>>m>>n>>k)
	{
		for(int i=0;i<=n;i++) G[i].clear();
		int xi,yi;
		for(int i=0;i<n;i++) {
			scanf("%d%d",&xi,&yi);
			vtype[xi]=yi;
		}
		for(int i=0;i<m;i++){
			scanf("%d%d",&xi,&yi);
			G[xi].push_back(yi);
			G[yi].push_back(xi);
		}
		memset(dis,0x3f,sizeof(dis));
		memset(num,0,sizeof(num));
		queue<Node> que;
		if(vtype[1]==0){
			dis[1][0]=0;
		    num[1][0]=1;
		    que.push(Node(1,0));
		}
		else{
			dis[1][1]=0;
			num[1][1]=1;
			que.push(Node(1,1));
		}
		while(!que.empty()){
			Node xx=que.front();que.pop();
			vis[xx.v][xx.k]=0;
			int u=xx.v; int kk=xx.k;
			for(int i=0;i<G[u].size();i++){
				int v=G[u][i];
				if(vtype[v]==0){
					if(vis[v][kk]){
						if(dis[u][kk]+1<dis[v][kk]){
							dis[v][kk]=dis[u][kk]+1;
							num[v][kk]=num[u][kk];
						}
						else if(dis[u][kk]+1==dis[v][kk]){
							num[v][kk]+=num[u][kk];
						}
					}
					else{
						if(dis[u][kk]+1<dis[v][kk]){
							dis[v][kk]=dis[u][kk]+1;
							num[v][kk]=num[u][kk];
							que.push(Node(v,kk));
							vis[v][kk]=1;
						}
						else if(dis[u][kk]+1==dis[v][kk]){
							num[v][kk]+=num[u][kk];
							que.push(Node(v,kk));
							vis[v][kk]=1;
						}
					}
				}
				else{
					if(kk==k) continue;
					if(vis[v][kk+1]){
						if(dis[u][kk]+1<dis[v][kk+1]){
							dis[v][kk+1]=dis[u][kk]+1;
							num[v][kk+1]=num[u][kk];
						}
						else if(dis[u][kk]+1==dis[v][kk+1]){
							num[v][kk+1]+=num[u][kk];
						}
					}
					else{
						if(dis[u][kk]+1<dis[v][kk+1]){
							dis[v][kk+1]=dis[u][kk]+1;
							num[v][kk+1]=num[u][kk];
							que.push(Node(v,kk+1));
							vis[v][kk+1]=1;
						}
						else if(dis[u][kk]+1==dis[v][kk+1]){
							num[v][kk+1]+=num[u][kk];
							que.push(Node(v,kk+1));
							vis[v][kk+1]=1;
						}
					}
				}
			}
		}
		int mdis=inf;
		for(int i=0;i<=k;i++){
			mdis=min(mdis,dis[n][i]);
		}
		if(mdis==inf) {
			puts("Impossible!");continue;
		}
		int ans=0;
		for(int i=0;i<=k;i++){
			if(dis[n][i]==mdis) ans+=num[n][i];
		}
		cout<<ans<<endl;
	}
	return 0;
}