题目链接点击打开链接
题意: 从n个数中选出k对数,使得每对数差的绝对值的平方和最小。
思路 dp[i][j] 表示从前i个数中选出j对的数的绝对值的差最小。
dp[i][j] = min (dp[i-1][j] , dp[i-2][j-1] + a[i-1]) ; i >= 2*j
dp[i][j] = INF i < 2*j
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string>
#include <string.h>
using namespace std;
const int maxn = 2000 + 4 ;
int dp[maxn][1002] , a[maxn];
int main()
{
//freopen("a.txt" , "r" , stdin) ;
int n , k ;
while(scanf("%d%d" , &n , &k) != EOF)
{
for(int i = 1 ; i <= n ; i ++)
scanf("%d" , &a[i]) ;
sort(a+ 1 , a+n+1);
for(int i = 1 ; i < n ; i++)
{
a[i] = a[i+1] - a[i];
a[i]*= a[i] ;
//printf("%d")
}
for(int i = 1 ; i <= n ; i ++)
dp[i][0] = 0 ;
for(int j = 1 ; j<= k ; j ++)
{
for(int i = j * 2 ; i <= n ; i ++)
{
if(i == j * 2) dp[i][j] = dp[i-2][j-1] + a[i-1] ;
else dp[i][j] = min (dp[i-1][j] , dp[i-2][j-1] + a[i-1]) ;
}
}
printf("%d\n" , dp[n][k]) ;
}
return 0;
}
dp[i][j] 表示从i到n中选出j对数
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string>
#include <string.h>
using namespace std;
const int maxn = 2000 + 4 , INF = 1e9 ;
int dp[maxn][1002] ;
int A[maxn] , visit[maxn] , n , k;
int rec (int i , int j)
{
if(dp[i][j] >= 0 ) return dp[i][j];
int res ;
int M = (A[i+1] - A[i])*(A[i+1] - A[i]) ;
if(j == 0) res = 0 ;
else if(n - i + 1 < 2*j) res = INF ;
else res = min(rec(i+1 , j) , rec(i+2 , j-1) + M) ;
return dp[i][j]=res ;
}
int main()
{
//freopen("a.txt" , "r" , stdin) ;
while(scanf("%d%d" , &n , &k) != EOF)
{
for(int i = 1 ; i <= n ; i ++)
scanf("%d" , &A[i] ) ;
sort(A+1 , A+n+1) ;
memset(dp , -1 , sizeof(dp)) ;
printf("%d\n" , rec(1,k)) ;
}
return 0;
}