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

题目描述:

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

解题思路:

首先分别求出数组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=0;k<len1;k++)
	{
		int t=avg-(sum1-arr[k]);//(avg+arr[k]-sum1)//平均值减去把1中交换数字剩余的值的结果就是brr中要交换的结果  
		it=se.find(t);//用迭代器接收在2中找到的值
		if(it!=se.end())
		{
			res[0]=arr[k];
			res[1]=t;
			break;
		}
	}
	return res;
}
int main()
{
	int m;
	cin>>m;
	vector<int> arr;
	for(int i=0;i<m;i++)
	{
		int tmp;
		cin>>tmp;
		arr.push_back(tmp);
	}
	int n;
	cin>>n;
	vector<int> brr;
	for(int i=0;i<n;i++)
	{
		int tmp1;
		cin>>tmp1;
		brr.push_back(tmp1);
	}
	vector<int> res=Change(arr,brr);
	for(int i=0;i<res.size();i++)
	{
		cout<<res[i]<<" ";
	}
	cout<<endl;		
}

	

运行结果: