5J - 复习时间

为了能过个好年，xhd开始复习了，于是每天晚上背着书往教室跑。xhd复习有个习惯，在复习完一门课后，他总是挑一门更简单的课进行复习，而他复习这门课的效率为两门课的难度差的平方,而复习第一门课的效率为100和这门课的难度差的平方。xhd这学期选了n门课，但是一晚上他最多只能复习m门课，请问他一晚上复习的最高效率值是多少？ 

Input

输入数据的第一行是一个数据T，表示有T组数据。 
每组数据的第一行是两个整数n(1 <= n <= 40)，m(1 <= m <= n)。 
接着有n行，每行有一个正整数a(1 <= a <= 100)，表示这门课的难度值。

Output

对于每组输入数据，输出一个整数，表示最高效率值。

Sample Input

2
2 2
52
25
12 5
89
64
6
43
56
72
92
23
20
22
37
31

Sample Output

5625
8836 

// 贪心算法
// 以后直接sort完事

 #include<stdio.h>

 void shell_sort(int a[], int len, int flag)
 {
     int i, j, k, gap, t;
     for(gap=len/; gap>; gap/=)
         for(i=gap; i<len; i++)
             for(j=i;j-gap>=&&a[j]<a[j-gap];j-=gap)
             { t=a[j-gap]; a[j-gap]=a[j]; a[j]=t; }
     if(flag)
         for(i=;i<len/;i++)
         { t=a[i]; a[i]=a[len--i]; a[len--i]=t; }
 }

 int main()
 {
     int t, n, m, a[], i;
     scanf("%d", &t);
     while(t--)
     {
         scanf("%d %d",&n, &m);
         for(i=;i<n;i++)
             scanf("%d", &a[i]);
         shell_sort(a,n,);
         printf("%d\n", (-a[])*(-a[]));
     }
     return ;
 }