BZOJ2797: [Poi2012]Squarks

BZOJ2797

将${a}$数组排序。
$a_1=x_1+x_2,a_2=x_1+x_3$
$x_2+x_3$可能为$a_1……a_n$
发现如果确定了$x_1,x_2,x_3$，在数组a中删去$x_1+x_2,x_1+x_3,x_2+x_3$
剩下的最小的就为$x_1+x_4$，那么$x_4$就确定了，以此类推，后面的就都可求了。
那么我们就只需要枚举$x_2+x_3$的值（在a数组中的位置）即可。
然后需要用一个支持插入，删除，储存重复值，查询最小值的东西维护。
$multiset$完美解决。
QAQ好菜啊$set$都不会用。。$multiset$基础操作都搞了半天。。

【代码】

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <set>
#define N 10005
using namespace std;
typedef long long ll;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,sum;
int a[45005],tmp[305],ans[305][305];
multiset<int>M;

void Solve(int x)
{
    M.clear();
    for(int i=3;i<=m;i++) M.insert(a[i]);
    tmp[1]=a[1]+a[2]-x>>1;
    tmp[2]=a[1]-tmp[1];
    tmp[3]=a[2]-tmp[1];
    if(tmp[1]<0||tmp[2]<0||tmp[3]<0) return;
    M.erase(M.find(x));
    for(int i=4;i<=n;i++)
    {
        tmp[i]=*M.begin()-tmp[1];
        if(tmp[i]<0) return;
        for(int j=1;j<i;j++) {
            if(M.find(tmp[j]+tmp[i])==M.end()) return;
            M.erase(M.find(tmp[j]+tmp[i]));
        } 
    } 
    for(int i=2;i<=n;i++) if(tmp[i]<=tmp[i-1]) return;
    sum++;
    for(int i=1;i<=n;i++) ans[sum][i]=tmp[i];
}

int main()
{
    n=read();m=(n-1)*n/2;
    for(int i=1;i<=m;i++) a[i]=read();
    sort(a+1,a+1+m);
    for(int i=3;i<=n;i++) if(i==3||a[i]!=a[i-1]) Solve(a[i]);
    printf("%d\n",sum);
    for(int i=1;i<=sum;i++,printf("\n"))
        for(int j=1;j<=n;j++) 
            printf("%d ",ans[i][j]);
    return 0;
}