NYOJ325zb的生日

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

01背包的变形但是会超时,将max函数改为三目运算符可以过

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 100010
using namespace std;
int dp[N];
int data[25];
int main()
{
	int n;
	while(~scanf("%d",&n))
	{
		int w=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&data[i]);
			w+=data[i];
		}
		int v=w/2;
		memset(dp,0,sizeof(dp));
		for(int i=1;i<=n;i++)
		{
			for(int j=v;j>=data[i];j--)
			{
				dp[j]=(dp[j]>dp[j-data[i]]+data[i]?dp[j]:dp[j-data[i]]+data[i]);
			}
		}
		printf("%d\n",abs(w-dp[v]*2));
	}
}

看了下oj平台的最优程序好像是搜索+DP

#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;  
}          

猜你喜欢

转载自blog.csdn.net/love20165104027/article/details/80156898
325