|
马上注册,结交更多好友,享用更多功能^_^
您需要 登录 才可以下载或查看,没有账号?立即注册
x
题目:
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
代码#include "stdio.h"
#include <stdlib.h>
#include <string.h>
//#define LLDmax 9223372036854775805
int compare(const void *a, const void *b)
{
return *(int*)a - *(int*)b;
}
long long int min(int a,int b)
{
if(a>b) return b;
else return a;
}
int main()
{
int n,k,i,j;
int arr[2001]={0},dp[2001][1001]={0};//dp[i][j]表示在当前物品i,取j对物品时最小疲劳值为 dp[i][j]
while(scanf("%d %d",&n,&k))
{
for(i=1;i<=n;i++)
{
scanf("%d",&arr[i]);
}
arr[0]=0;
qsort(arr, n, sizeof(int), compare); //升序
memset(dp, 0, sizeof(dp));
/*dp[0][0]=0;
for(i=0;i<=n;i++)
{
dp[i][0]=LLDmax;
}*/
for(i=2;i<=n;i++)
{
for(j=1;j<=k&&j<=i/2;j++)
{
dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+(arr[i]-arr[i-1])*(arr[i]-arr[i-1]));
}
}
printf("%d\n",dp[n][k]);
}
}
调试错误提示
您的代码是没有问题的,但在大部分普通的电脑上,这段代码可能会报错。
重点在于您声明 arr 以及 dp 数组的行,这一行你申请了大量的内存资源,而不巧的是,这是 main 函数,所以,这些内存空间会申请在栈中。
更不巧的是,因为您使用的计算机可能是一个普通的电脑,而正常电脑的栈资源只有 1M~2M,所以,arr 和 dp 的数组大小已经远远超过了 2M
您可以将数组尺寸缩小,但可能会影响代码逻辑,更稳妥的方式是将数组申请在其他的空间中(如全局变量),让我尝试对您的代码进行修改:#include "stdio.h"
#include <stdlib.h>
#include <string.h>
//#define LLDmax 9223372036854775805
int compare(const void *a, const void *b)
{
return *(int*)a - *(int*)b;
}
long long int min(int a,int b)
{
if(a>b) return b;
else return a;
}
int n,k,i,j;
int arr[2001]={0},dp[2001][1001]={0};//dp[i][j]表示在当前物品i,取j对物品时最小疲劳值为 dp[i][j]
int main()
{
while(scanf("%d %d",&n,&k))
{
for(i=1;i<=n;i++)
{
scanf("%d",&arr[i]);
}
arr[0]=0;
qsort(arr, n, sizeof(int), compare); //升序
memset(dp, 0, sizeof(dp));
/*dp[0][0]=0;
for(i=0;i<=n;i++)
{
dp[i][0]=LLDmax;
}*/
for(i=2;i<=n;i++)
{
for(j=1;j<=k&&j<=i/2;j++)
{
dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+(arr[i]-arr[i-1])*(arr[i]-arr[i-1]));
}
}
printf("%d\n",dp[n][k]);
}
}
全局变量的空间比栈的空间大的多,将一些空间较大的变量一般情况下没有问题。
|
|