试题 算法训练 无聊的逗
内存限制:256.0MB C/C++时间限制:1.0s Java时间限制:3.0s Python时间限制:5.0s
问题描述
逗志芃在干了很多事情后终于闲下来了,然后就陷入了深深的无聊中。不过他想到了一个游戏来使他更无聊。他拿出n个木棍,然后选出其中一些粘成一根长的,然后再选一些粘成另一个长的,他想知道在两根一样长的情况下长度最长是多少。
输入格式
第一行一个数n,表示n个棍子。第二行n个数,每个数表示一根棍子的长度。
输出格式
一个数,最大的长度。
样例输入
4
1 2 3 1
样例输出
3
数据规模和约定 n<=15
考虑到数据量并不大,所有可以直接求出所有子集,然后对子集进行处理,详细代码如下。代码适用于c++98编译器。
#include <iostream>
#include <vector>
#include <numeric>
#include <cmath>
using namespace std;
int maxLen = 0;
//求所有子集
vector<vector<int> > getSubsets(vector<int> &nums) {
vector<vector<int> > res(1);//初始化包含1个空的vector
for (int i=0;i<nums.size();i++) {
int n = res.size();
for (int j = 0; j < n; ++j) {
vector<int> subset = res[j];
subset.push_back(nums[i]);
res.push_back(subset);
}
}
return res;
}
void split(vector<int> &nums,int index,int addA,int addB){
if(addA == addB){
maxLen = max(maxLen,addA);
return;
}
if(index >= nums.size()) return;
//选择
split(nums,index+1,addA+nums[index],addB-nums[index]);
//不选择
split(nums,index+1,addA,addB);
}
int main() {
int n,num;
vector<int> nums = vector<int>();
// 输入集合
cin>>n;
for(int i=0;i<n;i++){
cin>>num;
nums.push_back(num);
}
// 获取所有子集
vector<vector<int> > res = getSubsets(nums);
for(int i=0;i<res.size();i++){
int size = res[i].size();
//数量小于2就跳过
if(size<2) continue;
int maxNum = 0,sum=0;
for(int j=0;j<res[i].size();j++){
sum += res[i][j];
maxNum = max(maxNum,res[i][j]);
}
//其他数之和小于最大值或者总和为奇数则分不了
if(sum-maxNum<maxNum || sum%2 != 0) continue;
else if(sum-maxNum==maxNum){//其他数之和正好等于最大值,直接分
maxLen = max(maxLen,maxNum);
}
else split(res[i],0,0,sum);//递归分配
}
cout<<maxLen;
return 0;
}