NYOJ zb的生日

zb的生日

时间限制： 3000 ms  |  内存限制： 65535 KB

难度： 2

描述

今天是阴历七月初五，acm队员zb的生日。zb正在和C小加、never在武汉集训。他想给这两位兄弟买点什么庆祝生日，经过调查，zb发现C小加和never都很喜欢吃西瓜，而且一吃就是一堆的那种，zb立刻下定决心买了一堆西瓜。当他准备把西瓜送给C小加和never的时候，遇到了一个难题，never和C小加不在一块住，只能把西瓜分成两堆给他们，为了对每个人都公平，他想让两堆的重量之差最小。每个西瓜的重量已知，你能帮帮他么？

输入

多组测试数据（<=1500）。数据以EOF结尾
第一行输入西瓜数量N (1 ≤ N ≤ 20)
第二行有N个数，W1, …, Wn (1 ≤ Wi ≤ 10000)分别代表每个西瓜的重量

输出

输出分成两堆后的质量差

样例输入

5
5 8 13 27 14

样例输出

3

动态规划解法 效率比较低 max必须宏定义 否则会超时

#include<stdio.h>
#include<string.h>
#define max(a, b) a > b ? a : b;
int a[25], dp[100005];
int main() {
  int n;
  while(scanf("%d", &n) != EOF) {
    int sum = 0;
    memset(dp, 0, sizeof(dp));
    for(int i = 0; i < n; i++) {
      scanf("%d", &a[i]);
      sum += a[i];
    }
    for(int i = 0; i < n; i++)
      for(int j = sum/2; j >= a[i]; j--)
        dp[j] = max(dp[j], dp[j-a[i]]+a[i])
    printf("%d\n", sum - 2*dp[sum/2]);
  }
} 

搜索解法 效率更快了 这个搜索比较简单

#include<stdio.h>
#include<stdlib.h>
int a[25], sum, min;
void dfs(int ans, int i) {
  if(i < 0) return ;
  int s = abs(sum-2*ans);
  if(s < min) min = s;
  dfs(ans+a[i], i-1);
  dfs(ans, i-1);
}
int main() {
  int n;
  while(scanf("%d", &n) != EOF) {
    sum = 0, min = 999999;
    for(int i = 0; i < n; i++) {
      scanf("%d", &a[i]);
      sum += a[i];
    }
    dfs(0, n-1); 
    printf("%d\n", min);
  }
}

贴下最优解

 
#include <stdio.h>
#define max(a,b) a>b?a:b
int V,ans,n,w[21],sum[21];
void dfs(int i,int cnt)
{
    if(i == 0)
    {
        ans = max(ans,cnt);
        return ;
    }
    if(ans == V || cnt+sum[i] <= ans)       //cut
        return ;
    if(cnt+w[i] <= V)
        dfs(i-1,cnt+w[i]);
    dfs(i-1,cnt);
}
int main()
{
    while(~scanf("%d",&n))
    {
        ans = 0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&w[i]);
            sum[i] = sum[i-1] + w[i];
        }
        V = sum[n]/2;
        dfs(n,0);
        printf("%d\n",sum[n]-2*ans);
    }
    return 0;
}