交换两个数组使得两个数组和相等

题目描述:

给定两个数组,交换数组中的两个元素,使得两个数组的元素和相等

解题思路:

首先分别求出数组1和数组2的和,开辟一个数组res将交换后的数字存进去,然后求出两个数组的平均值,平均值就是数组1和数组2交换之后数组中的和,在这里我用一个set将brr的值保存进去,然后遍历数组1,如果要把数组一中的值换掉,则数组一的和就变成sum1-arr[i],还差avg-(sum1-arr[i]),所以在数组二中找这个值,用一个迭代器返回之后,然后插入到res[1]中,交换的那个arr[i]就是res[0].返回res

代码:

#include<iostream>
#include<vector>
#include<set>
#include<algorithm>
using namespace std;

vector<int> Change(vector<int> arr,vector<int> brr)
{
	vector<int> res(2);
	int len1=arr.size();
	int len2=brr.size();
	int sum1=0;
	int sum2=0;
	set<int> se;
	set<int>::iterator it;
	for(int i=0;i<len1;i++)
	{
		sum1+=arr[i];//先求出数组1的和
	}
	for(int j=0;j<len2;j++)
	{
		sum2+=brr[j];//再求出数组2的和
		se.insert(brr[j]);//为了避免双层循环,把brr的数放入set中
	}
	int avg=(sum1+sum2)/2;
	for(int k=