【问题描述】
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
【思路解析】
神鬼题目想了我一个钟。从数学逻辑上很容易推理出:s[1]=a[1]+a[2],s[2]=a[1]+a[3],但是之后的 s[i] 就有很多种组合的可能了。
OK,现在我们知道a[1]+a[2],和a[1]+a[3],那就差一个a[2]+a[3]便可求出a[1]、a[2]、a[3],知道这三个数后便可推理出其他数:删除掉s数组中的a[1]+a[2],a[1]+a[3],a[2]+a[3],再找出s中的最小的数,这个数便是a[1]+a[4],推理出a[4]。推理出a[4]后,再在s数组中删除a[1]+a[4],a[2]+a[4],a[3]+a[4],再找最小元素,推出a[5]。
以此思路推理出所有的a[i]。
【AC代码】
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n;
int a[110];
int s[110];
bool vis[110];
int main()
{
while(cin>>n&&n!=0)
{
memset(a,0,sizeof(a));
memset(s,0,sizeof(s));
int num=n*(n-1)/2;
for(int i=1;i<=num;i++) cin>>s[i];
int cc=s[2]-s[1];//cc=a[3]-a[2]
for(int i=3;i<=num;i++)
{
//遍历可能的a[2]+a[3]
if((cc+s[i])%2) continue;
memset(vis,false,sizeof(vis));
bool flag=true;
vis[1]=vis[2]=vis[i]=true;
a[3]=(cc+s[i])/2;
a[1]=s[2]-a[3];
a[2]=s[1]-a[1];
for(int j=4;j<=n;j++)
{
int pos=0;
//找到现有的s中最小的 确定a[j]
for(int k=1;k<=num;k++)
{
if(vis[k]==false)
{
pos=k;
break;
}
}
if(pos)
{
vis[pos]=true;
a[j]=s[pos]-a[1];
//排除掉a[1]+a[j],a[2]+a[j],a[3]+a[j]~~
for(int k=1;k<j;k++)
{
int temp=a[j]+a[k];
bool ff=false;
for(int m=1;m<=num;m++)
{
if(s[m]==temp)
{
ff=true;
if(!vis[m])
{
vis[m]=true;
break;
}
}
}
if(!ff)
{
//如果s里面没有a[k]+a[j]这一元素
//就说明假设的这个a[2]+a[3]是错的
flag=false;
break;
}
}
//假设错误 立即跳出
if(!flag) break;
}
}
//假设正确 答案唯一 立即跳出
if(flag) break;
}
for(int i=1;i<=n;i++) cout<<a[i]<<" ";
cout<<endl;
}
return 0;
}
思维非常缜密,写代码的时候好几次把我脑袋写宕机了。