백준 11728 배열 합치기 Python

🐸 문제 정보

11728번: 배열 합치기

 

🤖 알고리즘

투포인터

 

⏱️ 풀이 시간

14.41m

 

📝 풀이

처음에 괜한 도전정신으로 힙으로 풀어보려다가 숫자 개수 보고 정신차리고 제대로 풀었다.

간단한 투포인터 문제이고, 두 리스트에 맨 처음부터 움직이는 인덱스를 두어 값을 비교하면서 하나씩 채워나가면 된다.

어차피 둘 다 정렬된 배열이기 때문에, 한 리스트에서 인덱스가 끝에 도착하면 나머지 리스트의 나머지 값은 그냥 더해주면 된다.

 

🧑‍💻 나의 답

# pypy3

import sys
input = sys.stdin.readline

n, m = list(map(int, input().rstrip().split()))
a = list(map(int, input().rstrip().split()))
b = list(map(int, input().rstrip().split()))

ai = 0
bi = 0
res = []

while (ai != n) and (bi != m):
    if a[ai] < b[bi]:
        res.append(a[ai])
        ai += 1
    else:
        res.append(b[bi])
        bi += 1

res += a[ai:] + b[bi:]
print(*res, sep=' ')