补: gym102823 / 2018CCPC桂林

全挂😂

A - Array Merge Gym - 102823A

题意:
有两个数组a,b，长度分别为n和m。现在需要将两个数组按顺序放入一个新的数组c，要求在 原数组内元素顺序不变的 前提下 。最小。
思路：
较大数尽量向前放；
较小数尽量向后放。

• （我的）错误思路：直接归并排序放，比较两数组当前元素，谁大放谁
• • 错误原因：可能a[i] 很小，但是a[i+1]非常大，但因为a[i]，会导致a[i+1]排到较后位置。

（大神）正确思路：
将a[1]和a[2]合并起来取平均值
也就是把a,b数组从前到后合并，每次新增一块，考虑末尾两块的合并，如果当前新增的块比末尾块的平均值要大，那么就把他们合并，这样保证a,b数组每一块的平均值是递减的

然后再按照归并排序的方式每次取大的就行了

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
 
const int maxl=1e5+10;
 
int n,m,cnta,cntb,cas;ll ans;
ll a[maxl],b[maxl];
struct node
{
	ll sum,num,mx;
}aq[maxl],bq[maxl];
 
inline bool cmp(const node &a,const node &b)
{
	return a.sum*b.num<=b.sum*a.num;
}
 
inline void prework()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	for(int i=1;i<=m;i++)
		scanf("%lld",&b[i]);
	cnta=0;cntb=0;node nd;
	for(int i=1;i<=n;i++)
	{
		nd=node{a[i],1,a[i]};
		while(cnta>0 && cmp(aq[cnta],nd))
			nd.sum+=aq[cnta].sum,nd.num+=aq[cnta].num,--cnta;
		aq[++cnta]=nd;
	}
	for(int i=1;i<=m;i++)
	{
		nd=node{b[i],1,b[i]};
		while(cntb>0 && cmp(bq[cntb],nd))
			nd.sum+=bq[cntb].sum,nd.num+=bq[cntb].num,--cntb;
		bq[++cntb]=nd;
	}
}
 
 
inline void mainwork()
{
	ans=0;int ida=0,idb=0,tot=0,tota=0,totb=0;
	for(int i=1;i<=cnta+cntb;i++)
	{
		if(ida==cnta || (idb<cntb&&cmp(aq[ida+1],bq[idb+1])))
		{
			++idb;
			for(int j=1;j<=bq[idb].num;j++)
				ans+=b[++totb]*(++tot);
			continue;
		}
		else
		{
			++ida;
			for(int j=1;j<=aq[ida].num;j++)
				ans+=a[++tota]*(++tot);
		}
	}
}
 
inline void print()
{
	printf("Case %d: %lld\n",++cas,ans);
}
 
int main()
{
	int t;
	scanf("%d",&t);
	for(int i=1;i<=t;i++)
	{
		prework();
		mainwork();
		print();
	}	
	return 0;
}