蓝桥杯算法训练-无聊的逗-集合

试题 算法训练 无聊的逗

内存限制：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;
}