ACM: HDU 1869 六度分离-Dijkstra算法

#include"algorithm"
#include"iostream"
#include"cstring"
#include"cstdlib"
#include"string"
#include"cstdio"
#include"vector"
#include"cmath"
#include"queue"
using namespace std;
typedef long long LL;
#define memset(x,y) memset(x,y,sizeof(x))
#define memcpy(x,y) memcpy(x,y,sizeof(x))
#define MX 401

int n,m;
const int dij_v=1005;
const int dij_edge=10005;

template <class T>
struct Dijkstra {
	struct Edge {
		int v,nxt;
		T w;
	} E[dij_edge<<1];

	int Head[dij_v],erear;
	T p[dij_v],INF;

	typedef pair< T ,int > PII;
	void edge_init() {
		erear=0;
		memset(Head,-1);
	}

	void edge_add(int u,int v,T w) {
		E[erear].v=v;
		E[erear].w=w;
		E[erear].nxt=Head[u];;
		Head[u]=erear++;
	}

	void run(int u) {
		memset(p,0x3f);
		INF=p[0];
		priority_queue<PII ,vector<PII >,greater<PII > >Q;
		while(!Q.empty()) {
			Q.pop();
		}
		Q.push(PII(0,u));
		p[u]=0;
		while(!Q.empty()) {
			PII a=Q.top();
			Q.pop();
			int u=a.second;
			if(a.first!=p[u])continue;
			for(int i=Head[u]; ~i; i=E[i].nxt) {
				int v=E[i].v;
				T w=E[i].w;
				if(p[u] + w <p[v]) {
					p[v]=w+p[u];
					Q.push(PII(p[v],v));
				}
			}
		}
		sort(p,p+n);
	}
};

Dijkstra<int > dij;

int main() {
	while(~scanf("%d%d",&n,&m)) {
		if(!n&&!m)break;
		dij.edge_init();
		for(int i=1; i<=m; i++) {
			int u,v;
			scanf("%d%d",&u,&v);
			dij.edge_add(u,v,1);
			dij.edge_add(v,u,1);
		}
		int flag=1;
		for(int i=0; i<n; i++) {
			dij.run(i);
			if(dij.p[n-1]>7)flag=0;
		}
		printf("%s\n",flag?"Yes":"No");
	}
	return 0;
}