问题描述
逗志芃在干了很多事情后终于闲下来了,然后就陷入了深深的无聊中。不过他想到了一个游戏来使他更无聊。他拿出n个木棍,然后选出其中一些粘成一根长的,然后再选一些粘成另一个长的,他想知道在两根一样长的情况下长度最长是多少。
输入格式
第一行一个数n,表示n个棍子。第二行n个数,每个数表示一根棍子的长度。
输出格式
一个数,最大的长度。
样例输入
4
1 2 3 1
样例输出
3
数据规模和约定
n<=15
思路
01背包问题,搜索状态
#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <vector>
#include <bits/stdc++.h>
using namespace std;
int get(vector<int> nums)
{
int n = nums.size();
if (n < 2)//n<2一定不成立
return 0;
int sum = accumulate(nums.begin()+1, nums.end(), 0);
int maxNum = *max_element(nums.begin()+1, nums.end());//获取最大木头长度,max_element返回位置
int target = sum / 2;//当前背包的大小
if (maxNum > target) return 0;//若最大木头大于背包大小,不成立;
vector<vector<int> > dp(n, vector<int>(target+1, 0));//创建背包
//初始化背包
for (int i = 0; i < n; i++)
{
for (int j = 0; j < target+1; j++)
{
dp[i][j] = 0;
}
}
//01背包模板
for (int i = 1; i < n; i++)
{
for (int j = 1; j <= target; j++)
{
if (j < nums[i])
{
dp[i][j] = dp[i - 1][j];
}
else
{
dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - nums[i]] + nums[i]);
}
}
}
return dp[n - 1][target] == sum / 2 ? sum / 2 : 0;
}
int main()
{
int n;
cin >> n;
int m = 0;
vector<int> a(n+1);
for (int i = 1; i <= n; i++)
{
cin >> a[i];
}
sort(a.begin()+1, a.end());//排序,方便处理数据
int sum = accumulate(a.begin()+1, a.end(),0);//计算木头总长,因为如果两个木头总和相等,那所有木头总和一定为偶数,若为奇数啧不成立
if (sum % 2 == 0)
{
m = get(a);
}
int i = 1;
while (i < a.size()) {
if ((sum - a[i]) % 2 == 0)
{
a.erase(a.begin() + i);//若总和为奇数,则先去掉一个数据,又因为要取最大的总和,所有从最小的开始去除
m = max(get(a), m);
i = 0;
continue;
}
sum = accumulate(a.begin() + 1, a.end(), 0);
i++;
}
cout << m;
return 0;
}