(Nowcoder) 牛客练习赛18 E 出题人的数组(数学)

视频题解戳这里

紧张 就是调了一手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;
}

猜你喜欢

转载自blog.csdn.net/TDD_Master/article/details/86556008
今日推荐