牛客OI赛制测试赛1 序列 ʕ •ᴥ•ʔ

小a有n个数，他想把他们划分为连续的权值相等的k段，但他不知道这是否可行。

每个数都必须被划分

这个问题对他来说太难了，于是他把这个问题丢给了你。

输入描述:

第一行为两个整数n,q，分别表示序列长度和询问个数。
第二行有n个数，表示序列中的每个数。
接下来的q行，每行包含一个数k，含义如题所示。

输出描述:

输出q行，每行对应一个数Yes或者No，分别表示可行/不可行

示例1

输入

5 3
2 1 3 -1 4
3
2
1

输出

Yes
No
Yes

备注:

对于的数据，

对于的数据，

对于的数据，

设ai表示数列中的第i个数，保证

保证数据完全随机

暴力 。。。。 不知道为什么当时自己没思路 。。

#include <cstring>
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <string.h>
//#include <map>
#define ll long long
using namespace std;
#define pai acos(-1,0)
ll a[100010];
int ans[100010];
ll n,q;
int find(int k,int t)
{
	int cnt=0;
	ll sum=0;
	for(int i=0;i<n;i++)
	{
		sum+=a[i];
		if(sum==k)
		sum=0,cnt++;
	}
	return cnt==t;
}
int main()
{
	
	memset(ans,0,sizeof(ans));
	scanf("%lld %lld",&n,&q);
	ll s=0;
	for(int i=0;i<n;i++)
	{
		scanf("%lld",&a[i]);
		s+=a[i];
	}
	for(int i=1;i<=n;i++)
	{
		if(s%i==0)
		{
			if(find(s/i,i))
			{
				ans[i]=1;
			}
			if(s/i<=n&&find(i,s/i))
			{
				ans[s/i]=1;
			}
		}
	}
	for(int i=1;i<=q;i++)
	{
		ll x;
		scanf("%lld",&x);
		if(x>n||!ans[x])
		printf("No\n");
		else
		printf("Yes\n");
	}
	return 0;
	
}