小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;
}