为了能过个好年,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完事
1 #include<stdio.h> 2 3 void shell_sort(int a[], int len, int flag) 4 { 5 int i, j, k, gap, t; 6 for(gap=len/2; gap>0; gap/=2) 7 for(i=gap; i<len; i++) 8 for(j=i;j-gap>=0&&a[j]<a[j-gap];j-=gap) 9 { t=a[j-gap]; a[j-gap]=a[j]; a[j]=t; } 10 if(flag) 11 for(i=0;i<len/2;i++) 12 { t=a[i]; a[i]=a[len-1-i]; a[len-1-i]=t; } 13 } 14 15 int main() 16 { 17 int t, n, m, a[100], i; 18 scanf("%d", &t); 19 while(t--) 20 { 21 scanf("%d %d",&n, &m); 22 for(i=0;i<n;i++) 23 scanf("%d", &a[i]); 24 shell_sort(a,n,0); 25 printf("%d ", (100-a[0])*(100-a[0])); 26 } 27 return 0; 28 }