题目描述:
给定两个数组,交换数组中的两个元素,使得两个数组的元素和相等
解题思路:
首先分别求出数组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=