18-kangrae-jo

[kangrae-jo]

🔗 문제 링크

[두 큐 합 같게 만들기]

✔️ 소요된 시간

49m

✨ 수도 코드

2022 KAKAO TECH INTERNSHIP Lv.2 문제 가져와봤습니다.

문제는 읽으면 바로 이해될거에요.
떡하니 두 큐 합 같게 만들기로 나와있어서 큐를 쓰려고 생각하고 빠르게 구현하려고 했습니다.

근데 뭔가 마음속에서 이런 생각이 들었습니다.
"그냥 벡터 그대로 인덱스를 옮겨가며 풀 수 있지 않을까?"

그래서 투 포인터 문제라고 생각하고 풀어봤습니다.

수도 코드는 다음과 같아요.

1. 두 벡터의 합을 구한다.

2. while (q1 != q2) 
2-1. q1 > q2 이면 q1에서 q2로 이동
2-2. q2 > q1 이면 q2에서 q1으로 이동
2-3. 포인터(index) 이동
2-4. 이동한 횟수++
2-5. 만약 이동한 횟수가 (두 벡터의 합) * 2 이상이면 return -1

3. return 이동한 횟수

20분 정도 지나고 구현이 끝나고 제출해봤는데.... 1번 케이스만 자꾸 틀리는겁니다....

    while (q1 != q2) {       
        if (q1 > q2) {
            q1 -= queue1[index1];
            q2 += queue1[index1];
            index1 = (index1 + 1) % len; // 여기 문제 있음!
        }
        else if (q1 < q2) {
            q2 -= queue2[index2];
            q1 += queue2[index2];
            index2 = (index2 + 1) % len; // 여기 문제 있음!
        }
        answer++;
        if (answer > len * 2 + 1) return -1; // 여기 문제 있음!
    }
    return answer;

20분 정도 더 고민했는데 도저히 틀린 이유를 모르겠어서 그냥 큐를 사용하는 방식으로 돌아갔습니다.
큐로 위 로직을 똑같이 구현하니 바로 정답이더라구요.

문제 풀이가 끝나고 생각을 좀 해봤습니다.
큐를 사용한 방식과 투 포인터를 사용한 방식의 차이는 인덱스 밖에 없다고 느꼈습니다.

그래서 디버깅을 좀 해봤는데 역시 인덱스가 문제더라구요.

index = (index + 1) % len;

이라는 코드로 큐를 문제없이 재현했다고 생각했는데, 그게 아니었습니다.

[1, 1]
[1, 5]

케이스가 위와 같을 때 문제가 발생합니다. ( 한쪽으로 몰리는 구조 )
(과정은 복잡하니 생략하고..... )

왜냐하면 한 쪽 포인터가 다시 0으로 돌아갔을 때, 다른 쪽 벡터로 이미 이동한 원소를 가르키기 때문입니다!
( 각자 한 번 생각해봐도 좋을 것 같아서.. 불친절한 설명 죄송합니다 )

그리고 이동한 횟수 즉 answer의 최대 값을 지정하는 것도 이 문제의 포인트라고 생각합니다.
저는 두 벡터의 크기의 합 * 2 라고 생각을 하는데요.
왜냐하면 각 큐가 다시 처음 상황으로 돌아온다면, 두 큐의 합을 같게 만들 수 없다고 생각하기 때문입니다!

📚 새롭게 알게된 내용

문제 풀이가 원소를 이동하는 방식이라면, 그냥 큐를 쓰자...