http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=1880
题意:给定n个数,要求分成两组,每组的个数最多相差1,求两组数的和相差最小的情况。
算法: 经典DP。
分析:用dp[i][j] 表示i个人的体重之和为j是否可能,显然转移应该从dp[i-1][j-?]来转移,但是
这里有一个问题,就是最后选进来的那个人如果设为k的话,因为没有规定i个人的范围,因此
这里的状态表示就没有“无后效性”了。 这里可以这么看,我们可以增加一个状态k,用dp[k][i][j]
表示前k个人中选取i个人,体重之和为j是否可能。则得到转移方程为:
dp[k][i][j] = dp[k-1][i][j] || dp[k-1][i-1][j-w[k]],即考虑第k个人取或者不取。
其实这个状态的得出也可以通过另外一个方面来得到,即:dp[i][j] 的求解的时候,k的for循环
放在最外层,这样就可以保证每个数只被考虑一次。
代码:
#include<stdio.h> #include<string.h> #include<stdlib.h> #define min(a,b) (a>b?b:a) #define max(a,b) (a>b?a:b) int n,sum; int w[101]; int dp[101][45001]; void DP() { memset(dp,0,sizeof(dp)); dp[0][0] = 1; for(int k=1;k<=n;k++) { for(int i=k;i>=1;i--) { for(int j=0;j<=sum;j++) { if(dp[i-1][j] == 1) { dp[i][j+w[k]] = 1; dp[i][j] = 1 ; } } } } int ave = n/2,res=0x7fffffff,min_,max_; for(int j=0;j<=sum;j++) { if(dp[ave][j]==1 && dp[n-ave][sum-j]==1) { if(res > abs(sum-2*j)){ res = abs(sum-2*j) ; min_ = min(j,sum-j); max_ = max(j,sum-j); } } } printf("%d %d\n",min_,max_); } int main() { while(scanf("%d",&n)!=EOF) { sum = 0 ; for(int i=1;i<=n;i++) { scanf("%d",&w[i]); sum += w[i] ; } DP(); } return 0; }