【OJ2726】集合问题，堆

2726:集合问题
查看 提交 统计 提问
总时间限制: 5000ms 内存限制: 65536kB
描述
有一组正整数，总数不超过1000，其中最大值记为M。现要将它们划分成N个集合，使得每个集合的元素之和与M的差的绝对值的和最小。

集合A中当前各元素之和记为SUM(A)，称为A的负荷；SUM(A)与M之差的绝对值称为A的负荷与理想负荷的偏差，简称为A的偏差。把这些整数划分成N个集合的方法是：按照从大到小的顺序，依次为每个整数分别选择一个集合；确定一个整数所属集合时，先计算各集合的负荷，将该整数分配给负荷最小的那个集合。

求使得各集合的偏差之和最小的划分方案中，集合的数目N。如果这样的方案不止一种，则输出各方案中，集合数最大的那种方案的集合数N。
输入
共输入K+1个整数。其中第一个整数是K代表要划分的整数总数，后面依次是K个整数的值。K不超过1000。
输出
一个整数，代表集合数N。
样例输入
8
2 4 9
12 16
80 28
72
样例输出
3
写在前面：人要是一急，控制变量名都会打错
——————————————————————————————————————————————
思路：集合数量直接1-n暴力枚举，最后算一下最小和可过
代码：

#include<cstdio>
#include<algorithm>
#include<queue>
#include<cmath>
using namespace std;
int n,a[1010],minn=0X7fffffff,ans=0;
bool flag;
priority_queue <int,vector<int>,greater<int> > q;
inline int comp(int x,int y)
{
    return x>y;
}
main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    sort(a+1,a+n+1,comp);
    for (int i=1;i<=n;i++)
    {
        int total=0;
        for (int j=1;j<=i;j++) q.push(0);
        for (int j=1;j<=n;j++)
        {
            int k=q.top();
            q.pop();
            q.push(k+a[j]);
        }
        for (int j=1;j<=i;j++) total+=abs(a[1]-q.top()),q.pop();
        if (minn>=total) ans=i,minn=total;
        else if (!flag) flag=1;
        else break;//一步小优化，不加也能过，加了能快一些，自己感受一下
    }
    printf("%d",ans);
}