O(NlgN)풀이로 AC를 받고 난 이후 O(N) 풀이가 궁금해서 @catniz님의 소스코드를 참고했습니다.

작동 원리를 이해하려고 여러 테스트케이스를 넣어보니 잘못된 정답이 나오는 테스트케이스가 꽤 많았습니다..

@catniz 님께는 죄송하지만 스페셜 저지가 잘못되지 않았나 생각됩니다.

잘못된 출력이 나오는 것들은 다음과 같은 케이스들입니다.

(입력1)

1

2 6

0 0 0 1 0 1

문제 조건에 따라 YES 1 0 0 1등을 출력해야하겠지만, YES 1 1 0 0의 출력 결과가 나왔습니다.

(입력2)

1

2 6

0 0 1 0 1 2

문제 조건에 따라 YES 1 2 1 등을 출력해야 하겠지만, YES 1 1 2의 출력 결과가 나왔습니다.