백준 2565번: 전깃줄 (Python)

문제

두 전봇대 A와 B 사이에 하나 둘씩 전깃줄을 추가하다 보니 전깃줄이 서로 교차하는 경우가 발생하였다. 합선의 위험이 있어 이들 중 몇 개의 전깃줄을 없애 전깃줄이 교차하지 않도록 만들려고 한다.

예를 들어, < 그림 1 >과 같이 전깃줄이 연결되어 있는 경우 A의 1번 위치와 B의 8번 위치를 잇는 전깃줄, A의 3번 위치와 B의 9번 위치를 잇는 전깃줄, A의 4번 위치와 B의 1번 위치를 잇는 전깃줄을 없애면 남아있는 모든 전깃줄이 서로 교차하지 않게 된다.

< 그림 1 >

전깃줄이 전봇대에 연결되는 위치는 전봇대 위에서부터 차례대로 번호가 매겨진다. 전깃줄의 개수와 전깃줄들이 두 전봇대에 연결되는 위치의 번호가 주어질 때, 남아있는 모든 전깃줄이 서로 교차하지 않게 하기 위해 없애야 하는 전깃줄의 최소 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에는 두 전봇대 사이의 전깃줄의 개수가 주어진다. 전깃줄의 개수는 100 이하의 자연수이다. 둘째 줄부터 한 줄에 하나씩 전깃줄이 A전봇대와 연결되는 위치의 번호와 B전봇대와 연결되는 위치의 번호가 차례로 주어진다. 위치의 번호는 500 이하의 자연수이고, 같은 위치에 두 개 이상의 전깃줄이 연결될 수 없다.

출력

첫째 줄에 남아있는 모든 전깃줄이 서로 교차하지 않게 하기 위해 없애야 하는 전깃줄의 최소 개수를 출력한다.

 

---

 

풀이

전깃줄 = (전봇대 1, 전봇대 2) N개에 대해, A(or B)를 기준으로 오름차순으로 정렬하고, 나머지에서 LIS를 찾으면 됨. 그러면, 제거해야하는 전기줄 수는 N-LIS가 된다.

인자 안에 값이 2개 이상 있을 때, 특정 값을 기준으로 정렬하는 알고리즘을 몰라서, selection sort(sorting도 다 까먹어서 가장 간단한 sort algorithm으로 짬)로 직접 정렬했는데 방법이 있더라. 일단은 내 코드:

 

코드

import sys

N = int(sys.stdin.readline())
A = [0]
B = [0]
dp = [0]*(N+1)
for i in range(N):
    a, b = map(int, sys.stdin.readline().split())
    A.append(a)
    B.append(b)

for i in range(1, N):
    min_index = i
    for j in range(i+1, N+1):
        if A[j] < A[min_index]:
            min_index = j
    A[i], A[min_index] = A[min_index], A[i]
    B[i], B[min_index] = B[min_index], B[i]
    
for i in range(N+1):
    for j in range(i):
        if B[i]>B[j]:
            dp[i] = max(dp[i], 1+dp[j])

print(N-max(dp))

 

피드백

다른 분의 코드를 조금 보니, 인자가 여러 개 있을 경우 특정 index에 대해서 내림차순/오름차순 정렬하는 방법이 있더라.

핵심: .sort(key=lambda x= x[i]) / list(k[1] for k in a)

참고: https://kingofbackend.tistory.com/98

 

코드 피드백

import sys

N=int(sys.stdin.readline())
A = [[0,0]] + [list(map(int, sys.stdin.readline().split())) for _ in range(N)] # 같은 인자끼리 묶기
# a = [[0,0], [1, 8], [3, 9], [2, 2], [4, 1], [6, 4], [10, 10], [9, 7], [7, 6]]
A.sort(key=lambda x: x[0]) # sort(key=lambda x: x[i])
# a = [[0,0], [1, 8], [2, 2], [3, 9], [4, 1], [6, 4], [7, 6], [9, 7], [10, 10]]
B = list(k[1] for k in A) # k = [1,8], [2, 2], ... 인데 여기서 1st index를 선택해줌.
# b = [0, 8, 2, 9, 1, 4, 6, 7, 10]

dp = [0]*(N+1)
for i in range(N+1):
    for j in range(i):
        if B[j] < B[i]:
            dp[i] = max(dp[j]+1, dp[i])
print(N-max(dp))