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);

}

}