Fork me on GitHub

题目1452:搬寝室(dp题目)

题目链接:http://ac.jobdu.com/problem.php?pid=1452

详解链接:https://github.com/zpfbuaa/JobduInCPlusPlus

参考代码:

//
//  1452 搬寝室.cpp
//  Jobdu
//
//  Created by PengFei_Zheng on 24/04/2017.
//  Copyright © 2017 PengFei_Zheng. All rights reserved.
//
 
#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <string.h>
#include <cmath>
#include <limits.h>
//#define MAX 0x7fffffff
#define MAX_SIZE 2001
#define MAX_ROW 1001
using namespace std;
 
int n, k;
 
int weight[MAX_SIZE];
long long dp[MAX_ROW][MAX_SIZE];
 
int main(){
    while(scanf("%d%d",&n,&k)!=EOF){
        for(int i = 1 ; i <= n ; i++){
            scanf("%d",&weight[i]);
        }
        sort(weight+1,weight+1+n);
        for(int i = 1 ; i <= n ; i++){
            dp[0][i]=0;
        }
        for(int i = 1 ; i <= k ; i++){
            for(int j = 2*i ; j <= n ; j++){
                if(j>2*i){
                    dp[i][j] = dp[i][j-1];
                }
                else
                    dp[i][j] = INT_MAX;
                 
                if(dp[i][j] > dp[i-1][j-2]+pow(weight[j]-weight[j-1],2)){
                    dp[i][j] = dp[i-1][j-2]+pow(weight[j]-weight[j-1],2);
                }
            }
        }
        printf("%lld\n",dp[k][n]);
    }
    return 0;
}
/**************************************************************
    Problem: 1452
    User: zpfbuaa
    Language: C++
    Result: Accepted
    Time:50 ms
    Memory:17176 kb
****************************************************************/

 

posted @ 2017-04-24 21:00  伊甸一点  阅读(279)  评论(0编辑  收藏  举报