第四章#4.5小希的数表

【问题描述】

Gardon 昨天给小希布置了一道作业，即根据一张由不超过 5000 的 N(3<=N<=100)个正整数组成的数表两两相加得到 N*(N-1)/2 个和，然后再将它们排序。例如，如果数表里含有四个数 1，3，4，9，那么正确答案是 4，5，7，10，12，13。小希做完作业以后出去玩了一阵，可是下午回家时发现原来的那张数表不见了，好在她做出的答案还在，你能帮助她根据她的答案计算出原来的数表么？

【输入形式】

包含多组数据，每组数据以一个 N 开头，接下来的一行有按照大小顺序排列的 N*(N-1)/2 个数，是小希完成的答案。文件最后以一个 0 结束。
假设输入保证解的存在性和唯一性。

【输出形式】

对于每组数据，输出原来的数表。它们也应当是按照顺序排列的。

【样例输入】

4
4 5 7 10 12 13
4
5 6 7 8 9 10
0

【样例输出】

1 3 4 9
2 3 4 6

#include<iostream>
using namespace std;
 
int N, n, key, sum[5000], ans[101], minn;
 
int look(int s)	//寻找第一个前缀为s且后缀不为key的sum[i]的i
{
	int low = 1, high = n, mid;
	while (low < high - 1)
	{
		mid = (low + high) / 2;
		if (sum[mid] / 1000>=s)high = mid;
		else low = mid;
	}
	if (sum[low] / 1000 != s)low = high;
	if (sum[low] / 1000 != s)return 0;
	while (sum[low] % 1000 == key && low < n && sum[low + 1] / 1000 == sum[low] / 1000)low++;
	if (sum[low] % 1000 == key)return 0;
	return low;
}
 
bool add(int d)//新增ans[d]，更新sum
{
	for (int i = 1; i<d; i++)
	{
		int k = look(ans[i] + ans[d]);
		if (k == 0)return false;
		sum[k] = sum[k] / 1000 * 1000 + key;
	}
	return true;
}
 
bool solve(int d)
{
	if (d > N)
	{
		for (int i = 1; i <= N; i++)cout << ans[i] << ((i < N) ? " " : "\n");
		return true;
	}
	while (sum[minn] % 1000 == key)minn++;
	ans[d] = sum[minn]/1000 - ans[1];
	if (!add(d))return false;
	return solve(d + 1);
}
 
int main()
{
	while (cin >> N)
	{
		if (N == 0)break;
		n = N*(N - 1) / 2;
		sum[0] = 0;
		for (int i = 1; i <= n; i++)
		{
			cin >> sum[i];
			sum[i] *= 1000;
		}
		for (key = 3; key <= N; key++)
		{
			if ((sum[1] + sum[2] - sum[key]) / 1000 % 2)continue;
			ans[1] = (sum[1] + sum[2] - sum[key]) / 2000;
			bool flag = true;
			for (int i = 2; i <= key && flag; i++)
			{
				ans[i] = sum[i - 1]/1000 - ans[1];
				if (!add(i))flag = false;
			}
			if (!flag)continue;
			minn = 1;
			if (solve(key+1))break;
		}
	}
	return 0;
}