最小K个数之和

描述

输入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

#include <stdio.h>
#include <string.h>

int main(){
    int flag[2001];
    int n;
    int k;
    int i;
    int number;
    int sum;
    
    while(scanf("%d%d",&n,&k)!=EOF){
        memset(flag,0,sizeof(int)*2001);
        
        for(i=0;i<n;i++){
            scanf("%d",&number);
            
            flag[number]++;
        }
        
        sum=0;
        for(i=1;i<=2000;i++){
            if(flag[i]!=0){
                if(k>flag[i]){
                    sum+=flag[i]*i;
                    k-=flag[i];
                }
                else{
                    sum+=k*i;
                    break;
                }
            }    
        }
        printf("%d\n",sum);
    }
    return 0;
}