试题 算法训练 娜神平衡 C++

资源限制

时间限制：1.0s 内存限制：256.0MB

问题描述

　　娜娜是一个特别可爱的女孩子，作为学神的她最近在情感方面出现了一点点小问题。
　　她暗恋的琦琦是一名学霸，他只喜欢长得漂亮和学习很好的女生。
　　娜娜学习确实很神，但是她在琦琦面前却总是表现不出平时的神力。
　　琦琦感受到了娜娜对他的爱，但是他还是觉得娜娜的学习并不是特别好，于是他出了一道题给娜娜。
　　“娜娜，我们之间的关系需要在不断深入的同时保持一定的平衡，不可以你总是强势或者我总是弱势。”
　　琦琦给了娜娜一些两两不等的数，希望娜娜能把这些数分成两组A和B，满足以下条件：
　　1：每一次只能操作一个数，即只取出一个数分入A中或B中；
　　2：每一次操作完成后，A中数之和与B中数之和的差不能超过r。
　　新时代的丘比特们啊，帮帮娜娜吧！

输入格式

　　输入共两行。
　　第一行包括两个正整数n和r，n表示琦琦一共给了n个数，r的意义见题目描述。
　　第二行包括n个正整数，分别表示琦琦给的n个数。

输出格式

　　输出共两行，分别把A与B两组数按从小到大输出。
　　注意输入中n个数的第一个必须分入A组。
　　琦琦保证这样的输出唯一。

样例输入

4 10
9 6 4 20

样例输出

4 6 9
20

样例说明

　　先把4和6先后分入A组，再把20分入B组，最后把9分入A组。

数据规模和约定

　　很小，真的很小。

主要思路：

先试着向A中放，如果放不了就向B中放，每放一个就把一个放置位置true，回退时如果是因为既不能向A又不能向B才回退就说明尝试是错误的，把元素拿出来并且同时将标志位归false。

缺陷：

总感觉自己先按照一个模糊的思路去走，然后发现错误后打补丁打的太多，肯定代码是不太好的，希望有大佬提出优化意见。

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
vector<int> A;
vector<int> B;
vector<bool> visit;
int n,r;
int balance(vector<int> nums,int counts,int sumA, int sumB);//递归函数，nums为传入数组，counts为已经放进去的数字的个数，sumA,sumB分别表示AB两个数组的和
int main()
{
	int i,j,temp,first;
	cin >> n >> r;
	vector<int> nums;
	for (i = 0; i < n; i++)
	{
		cin >> temp;
		nums.push_back(temp);
		visit.push_back(false);
	}
	//sort(nums.begin(), nums.end());
	balance(nums,0,0,0);
	first = nums[0];
	vector<int> ::iterator it = find(A.begin(), A.end(),first);
	if (it != A.end())//如果发现A中没有存第一个元素，说明在B当中，（虽然我也不明白怎么在插入的时候就保证第一个元素在A），就把AB两个数组swap一下
	{
		;
	}
	else
	{
		vector<int> C(A.begin(), A.end());
		A = B;
		B = C;
	}
	sort(A.begin(), A.end());//因为要从小到大输出所以排序一下
	sort(B.begin(), B.end());
	for (i = 0; i < A.size(); i++)
	{
		cout << A[i]<<" ";
	}
	cout << endl;
	for (i=0; i < B.size(); i++)
	{
		cout << B[i]<<" ";
	}
	return 0;
}
int balance(vector<int> nums,int counts,int sumA,int sumB)
{
	if (counts <= nums.size())//如果没有把全部元素放进AB
	{
		int i = 0,count=0,flag=0,full=0;
		while (i<nums.size())//从nums第一个元素开始寻找，因为可能存在后一个元素已经放入visit但是前一个没有放入的情况
		{
			while (visit[i] == true)//如果元素已经放入AB了就不断向后继续找元素
			{
				i++;
				if (i >= nums.size())//如果遍历完了所有元素都放进AB了就把标志满的标志变量置为一（为了退出大循环），并且退出此次循环
				{
					full = 1;
					break;
				}
			}
			if (full)//如果满了，就退出整个循环
			{
				break;
			}
			if (sumA + nums[i] - sumB <= r&&visit[i]==false)//如果该元素没有被放入AB中并且向A中放之后不会超过B r
			{
				A.push_back(nums[i]);//放入该元素
				visit[i] = true;//将该元素的标记位置为一，标志这个元素已经被放进去了
				flag=balance(nums, counts + 1,sumA+nums[i],sumB);//向下一层进行递归，如果是因为向着A或者B放元素均放不进去了，就返回0标志这是一次错误的尝试
				if (!flag)//错误的尝试
				{
					A.pop_back();//将刚刚放进去的元素拿出来
					visit[i] = false;//并且把该位置刚刚的记录抹除
				}
				else
				{
					count++;//如果正常放入就标记一次
				}
			}
			if (sumB + nums[i] - sumA <= r&&visit[i]==false)//当这个元素向着A放失败了，就尝试放进B里面去
			{
				B.push_back(nums[i]);
				visit[i] = true;
				flag=balance(nums, counts + 1,sumA,sumB+nums[i]);
				if (!flag)
				{
					B.pop_back();
					visit[i] = false;
				}
				else
				{
					count++;
				}
			}
			i++;//若是都失败了试试看下一个元素（如果完全成功那么就向后找所有的visit都应该是true了）
		}
		if (count == 0&&full==0)//如果既没有放入任何一个数组，同时nums也没有把所有的元素都成功放入，那么就认为这一次完全无法进行任何放置，说明上一步的放置（尝试）是错误的
		{
			return 0;
		}
		return 1;
	}
	else
	{
		return 1;//正常退出返回正常尝试
	}
}