视频题解戳这里
紧张 就是调了一手bug 丧
解题思路:直接单个考虑给答案的贡献,简单的将大的放前,直接归并是不对的,(比如A数组1,2222,2222,2222;B数组2,2,2,2),这个题目重新考虑是平均值的影响,直接戳视频吧。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
ll a[maxn],b[maxn],n,m;
struct node {
ll s;
ll num;
} ta[maxn],tb[maxn];
bool pd(node a,node b) { //a比b的平均值大返回true
return a.s*b.num > a.num*b.s;
}
int main() {
std::ios::sync_with_stdio(0);
cin>>n>>m;
int nn=0,mm=0;
for(int i=1; i<=n; ++i) {
cin>>a[i];
ta[++nn]=node {a[i],1};
while(nn>1) {
if(pd(ta[nn],ta[nn-1])) {
ta[nn-1].num+=ta[nn].num,ta[nn-1].s+=ta[nn].s;
nn--;
} else break;
}
}
for(int i=1; i<=m; ++i) {
cin>>b[i];
tb[++mm]=node {b[i],1};
while(mm>1) {
if(pd(tb[mm],tb[mm-1])) {
tb[mm-1].num+=tb[mm].num,tb[mm-1].s+=tb[mm].s;
mm--;
} else break;
}
}
ll i=1,j=1,cnt=1,xa=1,xb=1,ans=0;
while(i<=nn && j<=mm) {
if(pd(ta[i],tb[j])) { //a的均值大
for(int k=1; k<=ta[i].num; ++k) {
ans+=a[xa++]*(cnt++);
}
i++;
} else {
for(int k=1; k<=tb[j].num; ++k) {
ans+=b[xb++]*(cnt++);
}
j++;
}
}
while(i<=nn) {
for(int k=1; k<=ta[i].num; ++k) {
ans+=a[xa++]*(cnt++);
}
i++;
}
while(j<=mm) {
for(int k=1; k<=tb[j].num; ++k) {
ans+=b[xb++]*(cnt++);
}
j++;
}
cout<<ans<<endl;
return 0;
}