HDU1421,搬寝室,java准超时

搬寝室

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 17256    Accepted Submission(s): 5841

Problem Description

搬寝室是很累的,xhd深有体会.时间追述2006年7月9号,那天xhd迫于无奈要从27号楼搬到3号楼,因为10号要封楼了.看着寝室里的n件物品,xhd开始发呆,因为n是一个小于2000的整数,实在是太多了,于是xhd决定随便搬2*k件过去就行了.但还是会很累,因为2*k也不小是一个不大于n的整数.幸运的是xhd根据多年的搬东西的经验发现每搬一次的疲劳度是和左右手的物品的重量差的平方成正比(这里补充一句,xhd每次搬两件东西,左手一件右手一件).例如xhd左手拿重量为3的物品,右手拿重量为6的物品,则他搬完这次的疲劳度为(6-3)^2 = 9.现在可怜的xhd希望知道搬完这2*k件物品后的最佳状态是怎样的(也就是最低的疲劳度),请告诉他吧.

 

Input

每组输入数据有两行,第一行有两个数n,k(2<=2*k<=n<2000).第二行有n个整数分别表示n件物品的重量(重量是一个小于2^15的正整数).

 

Output

对应每组输入数据,输出数据只有一个表示他的最少的疲劳度,每个一行.

 

Sample Input

  
  

   
   2 1
1 3
  
  

 

Sample Output

  
  

   
   4
  
  

 

Author

xhd

 

Source

ACM暑期集训队练习赛（二）  

DP开始超时了，加了两点优化也只能准超时

import java.io.BufferedInputStream;
import java.util.Arrays;
import java.util.Scanner;
//差几十毫秒的准超时程序
public class P1421_ieayoio {
	public static void main(String[] args) {
		//不用BufferedInputStream会超时
		Scanner cin = new Scanner(new BufferedInputStream(System.in));
		while (cin.hasNext()) {
			int n = cin.nextInt();
			int k = cin.nextInt();
			int[] a = new int[n + 5];
			for (int i = 1; i <= n; i++) {
				a[i] = cin.nextInt();
			}
			Arrays.sort(a, 1, n + 1);//排序过后接近的重量将在一起
			int[][] f = new int[n + 5][n + 5];
			for (int i = 0; i < f.length; i++) {
				Arrays.fill(f[i], Integer.MAX_VALUE);
				f[i][0] = 0;//前i件物品被拿零对的疲劳
			}
			for (int i = 2; i <= n; i++) {
				for (int j = 1; j <= k && j * 2 <= i; j++) {//不加j * 2 <= i会超时
					//第j对不拿，或拿
					f[i][j] = Math.min(f[i - 1][j], f[i - 2][j - 1]
							+ (a[i] - a[i - 1]) * (a[i] - a[i - 1]));
				}
			}
			System.out.println(f[n][k]);

		}
	}

}