그냥 if문으로 비교하면 되는 문제다.
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
// Complete the angryProfessor function below. string angryProfessor(int k, vector<int> a) { array<string, 2> arrStr = {"YES", "NO"}; unsigned ret = 0; unsigned l = 0, m = 0, s = a.size(); for(int t : a) { if(t > 0) { m++; } else { l++; } if(s - m < k) { break; } else if(l >= k) { ret = 1; break; } } return arrStr.at(ret); } |
s - m < k
는 별 필요는 없으나 악의적인 배열이 올 경우를 위해 나도 모르게 넣고 말았다.