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<x2<x3..<xn是没问题的

那么两个数的和中最小的为x1+x2

次小的为x1+x3

现在考虑x2+x3的排名

比x2+x3小的一定是x1+xk

我们枚举x2+x3是哪一个值

可以解出x1,x2,x3

剩下的数中最小的为x1+x4

求出x4

同时可以算出x2+x4,x3+x4

除去已知的,剩下的数中最小x1+x5

依此类推

做法

我们考虑怎么找剩下的数

首先我们把所有的和插入set/heap里

每知道一个和就在里面删掉

考虑什么情况违法

x<0, xi-1==xi ，你求出来的和在原序列中不存在

solution

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cctype>
#include <algorithm>
#include <cmath>
#include <queue>
using namespace std;
const int M=307;

inline int rd(){
    int x=0;bool f=1;char c=getchar();
    for(;!isdigit(c);c=getchar()) if(c=='-') f=0;
    for(;isdigit(c);c=getchar()) x=x*10+c-48;
    return f?x:-x;
}

int n,m;
int a[M];
int b[M*M];

int ans[M][M],tot;

struct cmp{
    bool operator () (int x,int y){
        return x>y;
    }
};

priority_queue<int,vector<int>,cmp>q,ers;

void gao(int d){
    int tp=b[1]+b[2]+d;
    if(tp&1) return;
    tp/=2;
    a[3]=tp-b[1];
    a[2]=tp-b[2];
    a[1]=tp-d;

    int cnt;
    int i,j,val;

    while(!q.empty()) q.pop();
    while(!ers.empty()) ers.pop();
    for(i=1;i<=m;i++) q.push(b[i]);
    ers.push(b[1]);
    ers.push(b[2]);
    ers.push(d);

    for(cnt=4;cnt<=n;cnt++){
        while(!ers.empty()){
            if(ers.top()==q.top()){
                ers.pop();
                q.pop();
            }
            else break;
        }
        val=q.top();q.pop();
        a[cnt]=val-a[1];
        for(j=2;j<cnt;j++) ers.push(a[j]+a[cnt]);
    }

    while(!ers.empty()){
        if(ers.top()==q.top()){
            ers.pop();
            q.pop();
        }
        else break;
    }
    if(!ers.empty()) return;
    for(i=1;i<=n;i++) if(a[i]<=0) return;
    for(i=2;i<=n;i++) if(a[i]==a[i-1]) return;

    tot++;
    for(i=1;i<=n;i++) ans[tot][i]=a[i];
}

int main(){
    int i,j;
    n=rd();
    m=n*(n-1)/2;
    for(i=1;i<=m;i++) b[i]=rd();
    sort(b+1,b+m+1);

    for(i=3;i<=n;i++)
        if(i==3||b[i]!=b[i-1]) gao(b[i]);

    printf("%d\n",tot);
    for(i=1;i<=tot;i++){
        for(j=1;j<=n;j++) printf("%d ",ans[i][j]);
        puts("");
    }
}