C++ 상자 색

알고리즘

C++ 상자 색

Chars4785 2018. 12. 28. 15:16

n개의 색 주어지고 상자에 색칠

하지만 인접하게는 색칠 못함

입력

1 1 1 1 1 1

출력

입력

1 2 3 4 5 6 7 8

출력

YES

#include <iostream>

#include <vector>

using namespace std;

int color[1001];

int main() {

   int num;
   cin >> num;
   int sum=0;

   int array[num];

   for(int i =0;i<num;i++)
   {   
      cin >> array[i];
      
      if(color[array[i]]>=2)
         color[array[i]]=2;      
      else  
         color[array[i]]++;

   }

   if(num <6){

      cout<<"NO";
   
   }else{

      for(int i=1;i<num;i++)
      {
         sum+=color[i];
      }

      if(sum < 6)
         cout<<"NO";
      else
         cout<<"YES";

   }

   return 0;
    
}