[CF571B]Minimization(贪心+DP，好)

Posted 2022-01-03 coding-gaga

题目链接

http://codeforces.com/problemset/problem/571/B

题意

给数组，得到公式最小值。

题解

• 由题分成的子数组只有两种长度，每种长度的数组数量也是固定的。那么就是如何分，怎么计算目标值。
• 易得子数组内部的目标值等于最大值-最小值。
• 贪心在于原始数组排序，分出的一段段认为是直接由排序的原数组切割得到的，无交叉。想一想很有道理。
• 那么就是如何切割。DP即可。DP[i][j]表示i个长度1的子数组和j个长度2个子数组得到的目标值的最小值。DP[cntl1][cntl2]即是所求。
• 事实上，得到切割完的所有连续的子数组，是我们要拆开交错排列的数组。
• 最后要估计一下时间复杂度

代码

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

public class Main 
    public static void main(String args[]) 
        Scanner in=new Scanner(System.in);
        int n=in.nextInt();
        int k=in.nextInt();
        int[] arr=new int[n+1];
        for(int i=1;i<=n;++i) 
            arr[i]=in.nextInt();
        
        
        Arrays.sort(arr,1,n+1);//
        
        //两种长度的子数组长度和数量
        int cntLong=n%k;
        int cntShort=k-cntLong;
        int lenLong=n/k+1;
        int lenShort=n/k;
        
        //dp
        int[][] dp=new int[cntLong+1][cntShort+1];//
        for(int i=0;i<=cntLong;++i) //
            Arrays.fill(dp[i], Integer.MAX_VALUE);//
        
        dp[0][0]=0;
        
        for(int i=0;i<=cntLong;++i) //
            for(int j=0;j<=cntShort;++j) //
                if(i!=0) 
                    int idxStart=(i-1)*lenLong+j*lenShort+1;
                    int idxEnd=i*lenLong+j*lenShort;
                    dp[i][j]=Math.min(dp[i][j], dp[i-1][j]+arr[idxEnd]-arr[idxStart]);
                
                if(j!=0) 
                    int idxStart=i*lenLong+(j-1)*lenShort+1;
                    int idxEnd=i*lenLong+j*lenShort;
                    dp[i][j]=Math.min(dp[i][j], dp[i][j-1]+arr[idxEnd]-arr[idxStart]);
                
            
        
        System.out.println(dp[cntLong][cntShort]);//