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개씩 인덱스를 증가시키며 만족하는 값을 찾으면 된다.