[그리디 알고리즘] 카약과 강풍

카약과 강풍(2891)

 

---

 

[문제]

2890번을 보면 알겠지만, 상근이는 카약 대회를 개최했다. 그런데, 갑자기 엄청난 강풍이 경기장에 불었고, 일부 카약이 부서졌다. 경기는 5분 안에 시작해야 하는 상황이다.
다행히 일부 팀은 혹시 모를 사태에 대비해서 카약을 하나 더 경기장에 들고 왔다. 카약은 매우 무겁고 운반하기 어렵다. 따라서, 자신의 바로 다음이나 전에 경기하는 팀에게만 카약을 빌려주려고 한다. 즉, 팀 4는 여분의 카약을 3이나 5에게만 빌려줄 수 있다. 다른 팀에게서 받은 카약은 또 다른 팀에게 빌려줄 수 없다. 또, 카약을 하나 더 가져온 팀의 카약이 손상되었다면, 여분의 카약으로 경기에 출전하게되고, 이 카약은 다른 팀에게 빌려줄 수 없다.
카약이 부서진 팀과 하나 더 가져온 팀이 주어진다. 카약을 적절히 빌렸을 때 출발하지 못하는 팀의 최솟값은 몇 팀인지 구하는 프로그램을 작성하시오.

---

[입력]

첫째 줄에 팀의 수 N, 카약이 손상된 팀의 수 S, 카약을 하나 더 가져온 팀의 수 R이 주어진다. (2 ≤ N ≤ 10, 1 ≤ S, R ≤ N)
둘째 줄에는 카약이 손상된 팀의 번호가 주어진다. 팀 번호는 중복되지 않는다.
셋째 줄에는 카약을 하나 더 가져온 팀의 번호가 주어진다. 팀 번호는 중복되지 않는다.

---

[출력]

첫째 줄에 출발을 할 수 없는 팀의 최솟값을 출력한다.

---

[입력 예시]

5 2 1
2 4
3

---

[출력 예시]

1

---

[내 풀이]

 

#팀의 수, 카약이 손상된 팀의 수, 카약을 하나 더 가져온 팀의 수
n, s, r=map(int, input().split())
broken_kayak=list(map(int, input().split()))
more_kayak=list(map(int, input().split()))

team=[1] * n
#손상된 카약
for i in broken_kayak:
    team[i-1]-=1

#하나 더 가져온 카약
for i in more_kayak:
    team[i-1]+=1

#[1, 1, 1, 1, 1] -> [1, 0, 2, 0, 1]

#바로 전이나 다음에 경기하는 팀만 카약을 빌려줄 수 있다.
for i in range(n):
    #1번째 팀일 때
    if i==0:
        if team[i]==2 and team[i+1]==0:
            team[i]-=1
            team[i+1]+=1
    #마지막 팀일 때
    elif i==n-1:
        if team[i]==2 and team[i-1]==0:
            team[i]-=1
            team[i-1]+=1
    #그 외
    else:
        if team[i]==2 and team[i-1]==0:
            team[i]-=1
            team[i-1]+=1
        elif team[i]==2 and team[i+1]==0:
            team[i]-=1
            team[i+1]+=1

print(team.count(0))