티스토리 뷰

PS/BOJ Python

10819번 - 차이를 최대로

zpqmdh 2023. 10. 10. 12:13

https://www.acmicpc.net/problem/10819

 

10819번: 차이를 최대로

첫째 줄에 N (3 ≤ N ≤ 8)이 주어진다. 둘째 줄에는 배열 A에 들어있는 정수가 주어진다. 배열에 들어있는 정수는 -100보다 크거나 같고, 100보다 작거나 같다.

www.acmicpc.net

# 차이를 최대로
import sys
input = sys.stdin.readline

N = int(input())
arr = list(map(int, input().split()))
l = []
answer = -1e9

def solve(cnt):
    global answer
    if cnt == N:
        total = 0
        for i in range(cnt-1):
            total += abs(l[i]-l[i+1])
        answer = max(answer, total)
        return
    for i in range(N):
        if arr[i] not in l:
            l.append(arr[i])
            solve(cnt+1)
            l.pop()

solve(0)
print(answer)

처음에 문제를 푼 방법은 이랬다. 백트래킹으로 문제를 풀되, 방문체크를 그저 리스트 l에 원소가 있는지 없는지만 판단했다.

하지만 이는 같은 값의 데이터가 들어올 때는 l에 원소를 더할 수 없어 cnt와 N이 동일할 수가 없었다. 즉 total 값을 연산할 수 없기 때문에 answer는 -1e9에서 갱신이 되지 않는다.

 

반례 >>
6
20 1 15 1 4 10

 

따라서 visited 배열을 만들어 방문체크를 진행하여 아래처럼 코드를 수정하였더니 통과했다 :)

 

# 차이를 최대로
import sys
input = sys.stdin.readline

N = int(input())
arr = list(map(int, input().split()))
visited = [False for _ in range(N)]
l = []
answer = -1e9

def solve(cnt):
    global answer
    if cnt == N:
        total = 0
        for i in range(cnt-1):
            total += abs(l[i]-l[i+1])
        answer = max(answer, total)
        return
    for i in range(N):
        if visited[i] == False:
            visited[i] = True
            l.append(arr[i])
            solve(cnt+1)
            l.pop()
            visited[i] = False

solve(len(l))
print(answer)

'PS > BOJ Python' 카테고리의 다른 글

1120번 - 문자열  (0) 2023.10.12
10971번 - 외판원 순회 2  (1) 2023.10.11
1068번 - 트리  (0) 2023.10.06
11729번 - 하노이 탑 이동 순서  (0) 2023.10.04
14719번 - 빗물  (0) 2023.07.21
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2024/09   »
1 2 3 4 5 6 7
8 9 10 11 12 13 14
15 16 17 18 19 20 21
22 23 24 25 26 27 28
29 30
글 보관함