HDU 1421

搬寝室

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

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

//d[k][i][j] = min(k-1) + w[i][j];
#include <iostream>
#include <cstring>
#include <vector>
using namespace std;
const int N = 2000;
bool vis[N][N];
int f[N];
int d[100][100][100];
int fun(int a)//求平方 
{
    return a*a;
}
int main()
{
    int i,j,t,T;
    int n,k;
    while(cin>>n>>k)
    {
        //vector < vector<bool > > vis(n,vector<bool > n);
        memset(vis,false,sizeof(vis));
        memset(f,0,sizeof(f));
        memset(d,-1,sizeof(d));//不可为0，若两件物品重量相同，那就冲突啦 
        int min = 0x7fffffff;
        for(i=0;i<n;i++)
            cin>>f[i];
        for(i=0;i<n;i++)
            for(j=0;j<n;j++)
            if(i!=j)
            {
                int index = f[i] - f[j];
                d[0][i][j] = fun(index);
                //cout<<d[0][i][j]<<endl;
                if(d[0][i][j]<min)
                {
                    vis[i][j] = true;
                    min = d[0][i][j];
                }
            }
        for(t=1;t<=k;t++)
        for(i=0;i<n;i++)
            for(j=0;j<n;j++)
            if(!vis[i][j]&&d[t][i][j]!=-1)
            {
                int temp = f[i] - f[j];
                d[t][i][j] = min + fun(temp);
                if(d[t][i][j]<min)
                {
                    min = d[t][i][j];
                    vis[i][j] = true;
                }
            }
        cout<<min<<endl;
    }
    return 0;
}
                     
//且不说，三维数组d[1000][2000][2000]无法实现，这个思想就不对，感觉特别像贪心            
                
                
        

#include <iostream>
#include <cstring>
#include <cstdlib>
using namespace std;
int d[2010][1005];
int f[2002];
int fun(int a)//求平方 
{
    return a*a;
}
int min(int a,int b)
{
    return a>b?b:a;
}
int cmp(const void *a,const void *b)
{
    return *(int *)a - *(int *)b;
}
void dp(int n,int k)//n件物品曲k次 
{
    int i,j,t;
    for(i=1;i<=n;i++)
        d[i][0] = 0;
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=i/2;j++)
        {
            int index = f[i]-f[i-1];//不是j 
            if(i>2*j)//因为要对第i件物品选择状态，必须多出一件 
            {
                d[i][j] = min(d[i-1][j],d[i-2][j-1] + fun(index));//第i件物品要或者不要 
            }
            else
                d[i][j] = d[i-2][j-1] + fun(index);
        }
    }
    cout<<d[n][k]<<endl;
}
int main()
{
    int i,j,t,T;
    int n,k;
    while(cin>>n>>k)
    {
        memset(d,0,sizeof(d));
        memset(f,0,sizeof(f));
        for(i=1;i<=n;i++)
            cin>>f[i];
        qsort(f+1,n,sizeof(int),cmp);
        dp(n,k);
    }
    return 0;
}