fork(1) download
copy 
  1.    #include"stdio.h"
  2.    #include"iostream"
  3.    #include"stdlib.h"
  4.    #include"algorithm"
  5.  
  6. using namespace std;
  7.  
  8.  
  9. #define get(a) scanf("%lld",&a)
  10.  #define yes printf("yes\n")
  11.  #define no printf("no\n")
  12.  
  13.  
  14. int main()
  15. {
  16.     long long int t,n,tmp,p,i,j,k,a[22],sum,sump,itr;
  17.     get(t);
  18.     while(t--)
  19.     {
  20.         get(n);
  21.                 get(p);
  22.                 sum=0;
  23.         for(i=0;i<n;i++) {get(a[i]); sum+=a[i]; }
  24.         if(sum == 0)
  25.         {
  26.             if(p > n)
  27.                 no;
  28.             else
  29.                 yes;
  30.         }
  31.         else
  32.         {
  33.             if(sum % p != 0 || p>n)
  34.                 no;
  35.             else
  36.             {
  37.                 tmp=sum/p;
  38.                 itr=p;
  39.                 sort(a,a+n);
  40.                 while(itr)
  41.                 {
  42.                     sum=tmp;
  43.                     for(i=n-1;i>=0 && sum !=0 ;i--)
  44.                     {
  45.                         if(a[i]<=sum && a[i]!= 0)
  46.                         {
  47.                             sum-=a[i];
  48.                             a[i]=0;
  49.                         } 
  50.  
  51.                     }
  52.                     if(sum != 0)
  53.                         break;
  54.                     itr--;
  55.  
  56.                 }
  57.                 if(itr)
  58.                     no;
  59.                 else
  60.                 {
  61.                     for(i=0; i < n ; i++)
  62.                         if(a[i]!=0)
  63.                             break;
  64.                     if(i==n)
  65.                         yes;
  66.                     else
  67.                         no;
  68.                 } 
  69.  
  70.             }     
  71.  
  72.         }
  73.     }
  74.     return 0; 
  75.  
  76. }
  77.  
Success #stdin #stdout 0s 3304KB
comments (?)
stdin
copy 
1
7 2
4 9 10 17 17 19
stdout
copy 
no
https://ideone.com/AbX026
language:
C++ (gcc 8.3)
created:
9 years ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!