题目
链接:https://ac.nowcoder.com/acm/contest/11166/G
题意
给定长度n,再给定操作次数k,两个数组a[i],b[i]。每次操作可以交换a数组中的俩个元素。
求最大值
∑
i
=
1
n
\sum_{i=1}^n
∑i=1n(|ai-bi|)。
思路
对于ai和bi,它俩的绝对值为max(ai,bi)-min(ai,bi)。所以我们可以方便操作,让a是大的那个。具体实现就是,如果b>a,swap(a,b)。
再考虑俩组(ai,bi)和(aj,bj),由于前面的操作ai>bi,并且aj>bj,那么我们只需要考虑bi和aj的大小关系。
如果bi>aj,有
变换前|ai-bi|+|aj-bj|=ai-bi+aj-bj
交换后,为|aj-bi|+|ai-bj|=-aj+bi+ai-bj
两者的差值为,2*(bi-aj)
如果aj>bi,也有
变换前|ai-bi|+|aj-bj|=ai-bi+aj-bj
交换后,为|aj-bi|+|ai-bj|=aj-bi+ai-bj
差值为0,也就无效操作。
我们想让2*(bi-aj)的值最大,那么我们可以给a升序排列,给b降序排列,然后选出k对数。当然k不能超过n(没意思),多余的操作可以用第二种
情况解决,因为交换以后,差值为0。如果2*(bi-aj)的值小于0,那么直接退出就可以了。
源代码
#include <bits/stdc++.h>
#define inf 0x3f3f3f3f
#define ll long long
#define INF 0x7f7f7f7f
#define endl '\n'
#define mem(a, b) memset(a, b, sizeof(a))
#define open freopen("ii.txt", "r", stdin)
#define close freopen("oo.txt", "w", stdout)
#define IO \
ios::sync_with_stdio(false); \
cin.tie(0); \
cout.tie(0)
#define pb push_back
using namespace std;
typedef pair<ll, ll> PII;
const ll N = 5e5 + 10;
const double PI = 3.1415926535898;
const ll mod = 1e9 + 7;
using namespace std;
ll a[N],b[N];
void sovle()
{
ll n,k;
cin>>n>>k;
ll ans=0;
for(int i=1;i<=n;i++)
{
cin>>a[i];
}
for(int i=1;i<=n;i++)
{
cin>>b[i];
if(b[i]>a[i])swap(a[i],b[i]);//让a是大的,b是小的。
ans+=(a[i]-b[i]);
}
sort(a+1,a+1+n);//给a升序排列
sort(b+1,b+1+n,greater<ll>());//给b降序排列
for(int i=1;i<=k&&i<=n;i++)
{
ll tmp=b[i]-a[i];
tmp*=2;
if(tmp<0)break;//出现负数,说明后面的操作没有意义,直接退出
ans+=tmp;
}
cout<<ans<<endl;
}
int main()
{
int t;
t = 1;
while (t--)
{
sovle();
}
return 0;
}