Algorithm

[Algorithm]백준1448<python>(sil3)DP

Dave Song 2022. 6. 21. 15:49

삼각형 만들기 성공

 
시간 제한메모리 제한제출정답맞힌 사람정답 비율
2 초 128 MB 3976 1654 1325 44.809%

문제

세준이는 N개의 빨대를 가지고 있다. N개의 빨대 중에 3개의 빨대를 선택했을 때, 이 빨대로 삼각형을 만들 수 있다면, 세 변의 길이의 합의 최댓값을 구하고 싶다.

입력

첫째 줄에 빨대의 개수 N이 주어진다. N은 3보다 크거나 같고, 1,000,000보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에 빨대의 길이가 한 줄에 하나씩 주어진다. 빨대의 길이는 1,000,000보다 작거나 같은 자연수이다.

출력

첫째 줄에 삼각형 세 변의 길이의 합의 최댓값을 출력한다. 만약 삼각형을 만들 수 없으면 -1을 출력한다.

예제 입력 1 복사

3
1
2
3

예제 출력 1 복사

-1

예제 입력 2 복사

3
1
2
2

예제 출력 2 복사

5

예제 입력 3 복사

6
2
3
2
3
2
4

예제 출력 3 복사

10

예제 입력 4 복사

5
4
5
6
7
20

예제 출력 4 복사

18

<Code>

import sys
#from itertools import combinations as com
input = sys.stdin.readline

n = int(input())
arr= []
for i in range(n):
    arr.append(int(input()))

#k=list(com(arr,3))
arr.sort(reverse = True) #첫번째 원소 부터 내림차순 정렬
#print(k)

ans = -1
for i in range(len(arr)-2):

    if arr[i] < arr[i+1] + arr[i+2]: #삼각형 만들 수 있는 지
        ans = arr[i]+arr[i+1]+arr[i+2]
        break


print(ans)



# 메모리 초과 -> 조합

<Insight>

 

  • itertools 의 combination 함수를 사용하여 메모리 초과가 났다
  • greedy의 전형 적인 문제로 변의 둘레의 최대 값을 구하는 문제임으로 리스트를 내림차순으로 정렬하여 앞에서 3개씩 인덱스를 증가시키며 만족하는 값을 찾으면 된다.