bzoj 2797: [Poi2012]Squarks 数学

题意

设有n个互不相同的正整数{X1,X2,…Xn}，任取两个Xi,Xj(i≠j)，能算出Xi+Xj。
现在所有取法共n*(n-1)/2个和，要你求出X1,X2,…Xn。
3<=n<=300

分析

如果我们可以知道X1的话，就很容易推出后面的数了，那么考虑如何求出X1。
设那n*(n-1)/2个和从小到大排序后的数组为a1,a2,…
不难发现X1+X2=a1,X1+X3=a2，那如果能知道X2+X3的值是多少，就可以直接算出前三个数了。
再观察一下可以发现X2+X3的取值肯定是在a[3..n+1]中的，那就枚举是哪个值，然后把其他值算出来就好了。
时间复杂度$O(n^3logn)$。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<set>

const int N=305;

int n,m,a[N],b[N*N],ans[N][N],rk[N],tot;
std::multiset<int> se;
std::multiset<int>::iterator it;

bool cmp(int x,int y)
{
    return ans[x][1]>ans[y][1];
}

void check(int x)
{
    if (x>3&&b[x]==b[x-1]) return;
    if ((b[1]-b[2]+b[x])&1) return;
    se.clear();
    for (int i=1;i<=m;i++) se.insert(b[i]);
    a[2]=(b[1]-b[2]+b[x])/2;
    a[1]=b[1]-a[2];
    a[3]=b[2]-a[1];
    for (int i=1;i<3;i++)
        for (int j=i+1;j<=3;j++)
        {
            int w=a[i]+a[j];
            it=se.lower_bound(w);
            if (it==se.end()||(*it)!=w) return;
            se.erase(it);
        }
    for (int i=4;i<=n;i++)
    {
        it=se.begin();
        a[i]=(*it)-a[1];
        for (int j=1;j<i;j++)
        {
            int w=a[i]+a[j];
            it=se.lower_bound(w);
            if (it==se.end()||(*it)!=w) return;
            se.erase(it);
        }
    }
    tot++;
    for (int i=1;i<=n;i++) ans[tot][i]=a[i];
}

int main()
{
    scanf("%d",&n);m=n*(n-1)/2;
    for (int i=1;i<=m;i++) scanf("%d",&b[i]);
    std::sort(b+1,b+m+1);
    for (int i=3;i<=n+1;i++) check(i);
    for (int i=1;i<=tot;i++) rk[i]=i;
    std::sort(rk+1,rk+tot+1,cmp);
    printf("%d\n",tot);
    for (int i=1;i<=tot;i++)
    {
        for (int j=1;j<=n;j++) printf("%d ",ans[rk[i]][j]);
        puts("");
    }
    return 0;
}