codeforces 404C - Restore Graph

题目链接：http://codeforces.com/problemset/problem/404/C

题目大意：n个点，每个点上最多有m个点与其相连，某点到所有点的最短路，判断是否能还原这张图，如果可以输出相连的边。

题目分析：首先需要找到原点，如果一个点的距离是0的话，则是起点，但是肯定只有一个。然后判断一下其余距离的点是不是都是在[1,k]范围就行了。

代码参考：

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<stack>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<sstream>
#include<iostream>
#include<algorithm>
#include<functional>
using namespace std;
typedef long long LL;
const int N = 1 << 17;
vector<int> v[N];
int main()
{
	int n, m, i, j, k, x, y, z, r;
	while (~scanf("%d%d", &n, &k))
	{
		for (i = 0; i < N; ++i)//初始化
		{
			v[i].clear();
		}

		int M = -1;//求最远的距离

		for (i = 1; i <= n; ++i)
		{
			scanf("%d", &x);
			v[x].push_back(i);//保存第i个点的距离为x
			M = max(M, x);
		}

		if (v[0].size() != 1 || v[1].size() > k)//如果起点不唯一或者和起点的点数大于k
		{
			puts("-1");
			continue;
		}

		bool f = true;

		for (i = 2; i <= M; ++i)//枚举每个距离
		{
			if (v[i - 1].size() * (k - 1LL) < v[i].size())//每个点与前面的点相连之后还剩下k-1个位置可以和后面的点相连
			{//v[i - 1].size() * (k - 1LL)可能超出int范围
				puts("-1");
				f = false;
				break;
			}
		}

		if (f)
		{
			printf("%d\n", n - 1);//最小生成树由n-1条边连接

			for (i = 0; i < M; ++i)//枚举每个距离
			{
				int cnt = 1;//统计已经和几条边相连
				j = r = 0;

				while (j < v[i].size() && r < v[i + 1].size())
				{
					if (i > 0 && cnt < k)
					{
						printf("%d %d\n", v[i][j], v[i + 1][r]);
						cnt++;
						r++;
					}
					else if (i == 0 && cnt <= k)//i==0时特判
					{
						printf("%d %d\n", v[i][j], v[i + 1][r]);
						cnt++;
						r++;
					}
					else
					{
						j++;//如果超过k了，就将剩下的点接在另外的等价的点上
						cnt = 1;
					}
				}

			}
		}
	}

	return 0;
}