牛客OI赛制测试赛-序列-模拟

哇这道题好坑啊，可能是我太菜了

题意就是叫把一个连续序列分成K组，使得每个组的和都相等

我最开始的想法是由于要分成K组，那我们知道，每组一定有sum(a[i])/k这样我们只需要每次当num==sum/k时，把num变成0

这样我们看最后是不是0，即可判断是否可以分组，但是最后要考虑到末尾为0的情况，你比如，序列是9 -1 8 0，你分2组，那

么我们知道，在8的时候我们就把num清空了，这样flag遇到0就变变成负数，就判断错了，我们其实可以这样，在最后再次判断num==0如果还等于0，那么就令flag为1，否则为-1

感谢给我讲题的大佬

#include<iostream>
#include<string.h>
#include<stdio.h>
#include<algorithm>
using namespace std;
int main()
{
    int n,q;
    long long a[];
    while(~scanf("%d%d",&n,&q))
    {
        long long sum=;
        for (int i=; i<=n; i++)
        {
            scanf("%lld",&a[i]);
            sum+=a[i];
        }
        int k=;
        int cnt=;
        long long num=;
        for (int i=;i<=q;i++)
        {
            scanf("%d",&k);
            int flag=;
            if (sum%k)
            {
                flag=;
            }
            else
            {
                num=;
                cnt=;
                for (int i=; i<=n; i++)
                {
                    num+=a[i];
                     if (num==sum/k)
                    {
                        flag=;
                        num=;
                    }else {
                       flag=;
                    }
                }
                if (num==){
                    flag=;
                }
            }
            if (flag)printf("Yes\n");
                else printf("No\n");
        }
    }
    return ;
}