gym102823A. Array Merge 2018CCPC桂林

https://codeforces.com/gym/102823/problem/A

思维题还是太弱了。。。感觉思维短时间练不出效果，只能看看能不能积少成多

首先如果随便拿的话，那么肯定是大的放前面小的放后面

然而直接找着归并排序放的话显然就错了，因为可能a[1]很小，但是a[2]巨大无比，然而一直被a[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;
}