D. Divide and Summarize（codeforces #689 div2)

shejunda

于 2020-12-15 21:55:35 发布

阅读量210

点赞数

分类专栏： cf

本文链接：https://blog.csdn.net/shejunda/article/details/111241543

版权

cf 专栏收录该内容

12 篇文章 0 订阅

订阅专栏

这是一个关于算法的博客，主要探讨如何通过二分法解决一类特殊的数列操作问题。题目要求每次操作取数列的最大值和最小值的平均值，将数列分为两部分，并计算这两部分的和。博主提供了C++代码实现，通过递归的二分策略，查找所有可能的分段组合，并记录是否可以达到特定的和。在主函数中，博主演示了如何读取输入，调用解决方案并输出结果。该问题涉及数据结构、算法优化和递归技巧。

摘要由CSDN通过智能技术生成

题意：每次操作可以取数列的最大最小值的平均值，把数列分成大于和小于的两段（不改变原顺序），得到的数列的和是我们能得到的和，询问哪些数可以得到

题解：二分

#include<bits/stdc++.h>
using namespace std;

#define ll long long
const int maxn=1e5+5,inf=0x7fffffff;
int lcnt,rcnt,n;
int st1[maxn],st2[maxn],a[maxn];
map<ll,int> mp;

void solve(int l,int r)
{
	if(l==r){
		mp[a[l]]=1;
		return;
	}
	int maxd=n+1,mind=0,mid;
	for(int i=l;i<=r;i++){
		if(a[i]>a[mind])mind=i;
		if(a[i]<a[maxd])maxd=i;
	}
	int midd=(a[mind]+a[maxd])/2;
	lcnt=0; rcnt=0;
	for(int i=l;i<=r;i++){
		if(a[i]<=midd)st1[++lcnt]=a[i];
		else st2[++rcnt]=a[i];
	}
	int i=l;
	ll sum1=0,sum2=0;
	while(lcnt)a[i]=st1[lcnt],sum1+=a[i],i++,lcnt--;
	mid=i-1;
	while(rcnt)a[i]=st2[rcnt],sum2+=a[i],i++,rcnt--;
	mp[sum1]++;  mp[sum2]++;
	if(mid==l-1||mid==r)return;
	solve(l,mid); solve(mid+1,r);
}

int main()
{
	ll sum;
	int t,q,x;
	scanf("%d",&t);
	while(t--){
		sum=0;
		scanf("%d%d",&n,&q);
		for(int i=1;i<=n;i++)scanf("%d",&a[i]),sum+=a[i];
		mp[sum]=1;
		a[n+1]=inf;
		solve(1,n);
		while(q--){
			scanf("%d",&x);
			if(mp[x])printf("Yes\n");
			else printf("No\n");
		}
		mp.clear();
	}
	return 0;
}