nyoj——678 最小K个数之和

最小K个数之和

时间限制： 1000 ms  |  内存限制： 65535 KB

难度： 2

描述

输入n个整数，输出其中最小的K个数之和。例如输入4,5,1,1,6,2,7,3,3这9个数字，当k=4，则输出最小的4个数之和为7（1,1,2,3）。

输入

测试样例组数不超过10
每个测试案例包括2行：
第一行为2个整数n,k(1<=k<=n<=100000)
第二行包含n个整数，每个整数的范围为【1~2000】

输出

对应每个测试案例，输出最小的k个数之和。

样例输入

8 4
5 2 1 3 8 1 9 3

样例输出

7

来源

NYIST第一届校赛(专业组)

对所有数的排序思路和http://blog.csdn.net/wuxiushu/article/details/48178319相同，代码如下：

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#define N(n,m) memset(n,m,sizeof(n));
#define M(n,m,k) for(int n=m;n<k;n++)

using namespace std;

int a[2001];

int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        N(a,0)
        int x;
        M(i,0,n)
        {
            cin>>x;
            ++a[x];
        }
        int sum=0;
        M(i,1,2001)
        if(a[i]!=0)
        {
            if(a[i]>=m)
            {
                sum+=m*i;
                break;
            }
            else
            {
                sum+=a[i]*i;
                m-=a[i];
            }
        }
        printf("%d\n",sum);
    }
}