动态规划-筷子-Java

题目描述

A先生有很多双筷子。确切的说应该是很多根，因为筷子的长度不一，很难判断出哪两根是一双的。这天，A先生家里来了K个客人，A先生留下他们吃晚饭。加上A先生，A夫人和他们的孩子小A，共K+3个人。每人需要用一双筷子。A先生只好清理了一下筷子，共N根，长度为T1,T2,T3,……,TN.现在他想用这些筷子组合成K+3双，使每双的筷子长度差的平方和最小。(怎么不是和最小？？这要去问A先生了，呵呵)

---

输入

有两行，第一行为两个用空格隔开的整数，表示N,K(1≤N≤100, 0<k)；

输出

仅一行。如果凑不齐K+3双，输出-1，否则输出长度差平方和的最小值。

样例输入

10 1
1 1 2 3 3 3 4 6 10 20

样例输出

5

---

解析

1. 设数组a[N+1]表示N跟筷子的长度，s[i][j]表示筷子i和筷子j长度差的平方，dp[i][j]表示从前i根筷子里选出j双筷子的平方和的最小值；
2. 将a数组排序；
3. 初始化：dp[i][0] = 0; (0<=i<=N) (选择0双)，dp[i][j] = Integer.MAX_VALUE;(1<=i<=N, 1<=j<=K)；
4. 从第二根筷子开始遍历(因为一根凑不齐一双呀)，dp[i-1][j] 表示不选当前这跟筷子的情况，dp[i-2][j-1]+dp[i-1][i] 表示选择了这根筷子的情况(算平方差最小，一定是相邻的筷子，为什么不是i+1和i比较，因为我们目前为止只算到了第i根筷子，还不存在第i+1根筷子)；
5. 由此，我们得出了转移方程：dp[i][j] = Math.min(dp[i-1][j], dp[i-2][j-1]+s[i-1][i])；
6. 最后输出值dp[N][K+3]；

---

代码

import java.util.Arrays;
import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
    	Scanner sca = new Scanner(System.in);
    	
    	int N = sca.nextInt();
    	int K = sca.nextInt();
    	
    	int a[] = new int[N+1];
    	for(int i = 1; i<=N;i++)
    		a[i] = sca.nextInt();
    	sca.close();
    	
    	K += 3;
    	Arrays.sort(a);
    	int[][] s = new int[N+1][N+1]; //前后两根筷子的平方差
    	for(int i = 2;i <= N;i++)
    		s[i-1][i] = (a[i] - a[i-1])*(a[i] - a[i-1]);
    	
    	int[][] dp = new int[N+1][K+1]; //前i根筷子选出j双的最小平方差和
    	for(int i = 0;i<=N;i++)
    		dp[i][0] = 0;
    	for(int i = 1; i<=N;i++)
    		for(int j = 1;j<=K;j++)
    			dp[i][j] = Integer.MAX_VALUE;
    	
    	for(int i = 2;i<=N;i++) { //运用转移方程
    		for(int j = 1;j<=K;j++) {
    			if(j*2 <= i)
    				dp[i][j] = Math.min(dp[i-1][j], dp[i-2][j-1]+s[i-1][i]);
    			else
    				break;
    		}
    	}
    	
    	System.out.println(dp[N][K]);
    }
   
  
}

如果有写的不对的地方，敬请指正。

参考链接: (https://blog.csdn.net/clove_unique/article/details/47951133).