hdu3440 House Man 差分约束系统 SPFA算法

http://acm.hdu.edu.cn/showproblem.php?pid=3440

题意：现在有n幢房子，具有不同的高度。超人只能从矮的地方飞到刚好比当前房子高的地方。现在你可以移动房子，但是房子的相对位置不能改变，且超人每次跳跃的水平距离小于d，求最矮的房子到最高的房子的水平距离的最大值。

列出两种不等式：

①房子的相对位置不能改变：dis[i]-dis[i-1]>=1->dis[i-1]-dis[i]<-1

②超人每次跳跃的水平距离小于d：dis[i]-dis[j]<=d(h[i]>h[j]，表示从j房跳到i房)

代码：

#include<set>
#include<map>
#include<stack>
#include<queue>
#include<vector>
#include<string>
#include<bitset>

#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<ctime>

#include<iomanip>
#include<iostream>

#define debug cout<<"aaa"<<endl
#define d(a) cout<<a<<endl
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))
#define LL long long
#define lson l,mid,root<<1
#define rson mid+1,r,root<<1|1
#define MIN_INT (-2147483647-1)
#define MAX_INT 2147483647
#define MAX_LL 9223372036854775807i64
#define MIN_LL (-9223372036854775807i64-1)
using namespace std;

const int N = 1000 + 5;
const int mod = 1000000000 + 7;
const double eps = 1e-8;
int head[N],len,n,m;
int dis[N],cnt[N];
bool vis[N];

struct node{
	int to,next,w;
}edge[N*N];

void add(int i,int j,int w){
	edge[len].to=j;
	edge[len].w=w;
	edge[len].next=head[i];
	head[i]=len++;
} 

void init(){
	mem(head,-1),len=0;
}

int SPFA(int s,int t,int n,int head[],node edge[],int dis[]){
	//vis标记是否在队列中 
	mem(vis,0),mem(cnt,0);
	for(int i=0;i<=n;i++){
		dis[i]=MAX_INT;
	}
	dis[s]=0;
	queue<int> q;
	q.push(s);vis[s]=1;cnt[s]=1;
	while(!q.empty()){
		//队头元素出队，并且消除标记 
		int x=q.front();q.pop();vis[x]=0;
		for(int k=head[x];k!=-1;k=edge[k].next){
			int y=edge[k].to;
			if(dis[x]+edge[k].w<dis[y]){
				dis[y]=dis[x]+edge[k].w;//松弛
				if(!vis[y]){//点y不在队内 
					vis[y]=1;//标记 
					cnt[y]++;//统计次数
					q.push(y);
					if(cnt[y]>n)//超过入队次数上限说明有负环 
						return -1; 
				} 
			}
		} 
	}
	return dis[t]; 
}

struct Node{
	int id,h;
}a[N];

bool cmp(Node a,Node b){
	return a.h<b.h;
}

int main(){
	int n,u,v,d,tt,cas=0,s,t;
	scanf("%d",&tt);
	while(tt--){
		init();
		scanf("%d%d",&n,&d);
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i].h);
			a[i].id=i;
			if(i!=n){
				add(i+1,i,-1);
			}
		}
		sort(a+1,a+n+1,cmp);
		for(int i=1;i<n;i++){
			u=a[i].id,v=a[i+1].id;
			if(u>v) swap(u,v);
			add(u,v,d);
		}
		s=a[1].id,t=a[n].id;
		if(s>t) swap(s,t);
		printf("Case %d: %d\n",++cas,SPFA(s,t,n,head,edge,dis));
	}
	return 0;
}