Hakase and Nano HDU

http://acm.hdu.edu.cn/downloads/CCPC2018-Hangzhou-ProblemSet.pdf

#include <stdio.h>
#include <string.h>

int a[1000005];

int main()
{
    int t, i, n, d, x;
    scanf("%d", &t);
    while(t--)
    {
        x = 0;
        scanf("%d %d", &n, &d);
        for(i=0; i<n; i++)
        {
            scanf("%d", &a[i]);
            if(a[i]==1) x++;
        }
        if(d==1)
        {
            if(n%3==0&&x==n) printf("No
");
            else printf("Yes
");
        }
        else
        {
            if((x==n&&n%3==1) || (x==n-1&&n%3==1) || (x==n-1&&n%3==0))
                printf("No
");
            else printf("Yes
");
        }
    }
    return 0;
}