【noip模拟题】藏宝图（prim）

好神的一题。。

一开始没想多久就看题解了QAQ。。

首先我们发现，这棵树任意两个点的边一定是最小的（即所有其它这两个点的路径都比这条边大，才有可能出解）

然后生成树后再算距离判断即可。。

注意特判n=1.。。。。。。。。。。。。。。。。。

还有出题人一点都不良心。。。说好的0<=dist[i][j]<=10^9呢。。。搞得我爆了好几次。。。。。

#include <cstdio>
#include <cstring>
#include <cmath>
#include <string>
#include <iostream>
#include <algorithm>
#include <queue>
#include <set>
#include <vector>
#include <map>
using namespace std;
typedef long long ll;
#define pii pair<int, int>
#define mkpii make_pair<int, int>
#define pdi pair<double, int>
#define mkpdi make_pair<double, int>
#define pli pair<ll, int>
#define mkpli make_pair<ll, int>
#define rep(i, n) for(int i=0; i<(n); ++i)
#define for1(i,a,n) for(int i=(a);i<=(n);++i)
#define for2(i,a,n) for(int i=(a);i<(n);++i)
#define for3(i,a,n) for(int i=(a);i>=(n);--i)
#define for4(i,a,n) for(int i=(a);i>(n);--i)
#define CC(i,a) memset(i,a,sizeof(i))
#define read(a) a=getint()
#define print(a) printf("%d", a)
#define dbg(x) cout << (#x) << " = " << (x) << endl
#define error(x) (!(x)?puts("error"):0)
#define printarr2(a, b, c) for1(_, 1, b) { for1(__, 1, c) cout << a[_][__]; cout << endl; }
#define printarr1(a, b) for1(_, 1, b) cout << a[_] << '\t'; cout << endl
inline const ll getint() { ll r=0, k=1; char c=getchar(); for(; c<'0'||c>'9'; c=getchar()) if(c=='-') k=-1; for(; c>='0'&&c<='9'; c=getchar()) r=r*10+c-'0'; return k*r; }
inline const int max(const int &a, const int &b) { return a>b?a:b; }
inline const int min(const int &a, const int &b) { return a<b?a:b; }

const int N=2505, oo=~0u>>1;
int n, vis[N], from[N], ihead[N], cnt;
ll sum[N], child[N], mp[N][N], mp2[N][N], d[N];
struct dat { int next, to; ll w; }e[N<<1];
priority_queue<pli, vector<pli>, greater<pli> >q;
void add(int u, int v, ll w) {
	e[++cnt].next=ihead[u]; ihead[u]=cnt; e[cnt].to=v; e[cnt].w=w;
	e[++cnt].next=ihead[v]; ihead[v]=cnt; e[cnt].to=u; e[cnt].w=w;
}
void prim() {
	for1(i, 1, n) d[i]=oo;
	d[1]=0; q.push(mkpli(0, 1));
	while(!q.empty()) {
		int u=q.top().second; q.pop(); if(vis[u]) continue; vis[u]=1;
		for1(v, 1, n) if(!vis[v] && mp[u][v] && d[v]>mp[u][v]) {
			d[v]=mp[u][v];
			from[v]=u;
			q.push(mkpli(d[v], v));
		}
	}
	for1(i, 2, n) if(d[i]!=oo) add(from[i], i, d[i]);
}
void dfs(int x, int fa) {
	for1(i, 1, n) if(vis[i]) mp2[i][x]=mp2[x][i]=mp2[i][fa]+mp2[fa][x];
	int y; child[x]=0; sum[x]=0; vis[x]=1;
	for(int i=ihead[x]; i; i=e[i].next) if((y=e[i].to)!=fa) {
		mp2[x][y]=mp2[y][x]=e[i].w;
		dfs(y, x);
		sum[y]+=e[i].w;
		sum[x]+=e[i].w;
		child[y]++;
		child[x]++;
	}
}
void init() {
	for1(i, 1, n) vis[i]=ihead[i]=from[i]=child[i]=sum[i]=0;
	for1(i, 1, n) for1(j, 1, n) mp2[i][j]=mp[i][j]=0;
	cnt=0;
	while(!q.empty()) q.pop();
}
int main() {
	int cs=getint();
	while(cs--) {
		read(n); if(n==1) { puts("Yes"); puts("1"); n=getint(); continue; }
		init();
		int flag=0;

		for1(i, 1, n) for1(j, 1, n) read(mp[i][j]);
		prim();
		for1(i, 1, n) if(d[i]==oo) { puts("No"); flag=1; break; }
		if(flag) continue;

		for1(i, 1, n) vis[i]=0;
		dfs(1, 0);

		//for1(i, 1, n) { for1(j, 1, n) printf("%d ", mp2[i][j]); puts(""); }
		for1(i, 1, n) { for1(j, 1, n) if(mp2[i][j]!=mp[i][j]) { flag=1; break; } if(flag) break; }
		if(!flag) {
			double mx=-1e199; int ans;
			// for1(i, 1, n) dbg(sum[i]);
			// for1(i, 1, n) dbg(child[i]);
			for1(i, 1, n) if((double)sum[i]/(double)child[i]>mx) { mx=sum[i]/child[i]; ans=i; }
			printf("Yes\n%d\n", ans);
		}
		else puts("No");
	}
	return 0;
}

　　

---

背景

Czy爬上黑红树，到达了一个奇怪的地方……

题目描述

Czy发现了一张奇怪的藏宝图。图上有n个点，m条无向边。已经标出了图中两两之间距离dist。但是czy知道，只有当图刚好又是一颗树的时候，这张藏宝图才是真的。如果藏宝图是真的，那么经过点x的边的边权平均数最大的那个x是藏着宝物的地方。请计算这是不是真的藏宝图，如果是真的藏宝之处在哪里。

格式

输入数据第一行一个数T,表示T组数据。

对于每组数据，第一行一个n，表示藏宝图上的点的个数。

接下来n行,每行n个数，表示两两节点之间的距离。

输出一行或两行。第一行”Yes”或”No”，表示这是不是真的藏宝图。

若是真的藏宝图，第二行再输出一个数，表示哪个点是藏宝之处。

样例输入

2

3

0 7 9

7 0 2

9 2 0

3

0 2 7

2 0 9

7 9 0

样例输出

Yes

1

Yes

3

样例解释:第一棵树的形状是1--2--3。1、2之间的边权是7，2、3之间是2。

第二棵树的形状是2--1--3。2、1之间的边权是2，1、3之间是7。

数据范围

对于30%数据，n<=50,1<=树上的边的长度<=10^9。

对于50%数据，n<=600.

对于100%数据，1<=n<=2500,除30%小数据外任意0<=dist[i][j]<=10^9,T<=5