본문 바로가기

PS/프로그래머스

[프로그래머스 LV 2 ] 최솟값 만들기 파이썬/python

728x90

https://programmers.co.kr/learn/courses/30/lessons/12941

 

코딩테스트 연습 - 최솟값 만들기

길이가 같은 배열 A, B 두개가 있습니다. 각 배열은 자연수로 이루어져 있습니다. 배열 A, B에서 각각 한 개의 숫자를 뽑아 두 수를 곱합니다. 이러한 과정을 배열의 길이만큼 반복하며, 두 수를 곱

programmers.co.kr

 

그냥 정렬해서 풀면됨... 최솟값음

A의 가장 작은수랑 B의 가장 큰수를 곱하는 방식으로 식을 짜면됨

sort로 정렬해서 풀면되는데 바보같이 순열을 구해서 완전탐색을 돌림

 

심지어 완전탐색이 안돼서 메모이제이션도씀 ㅋㅋㅋㅋ 풀릴리가.......

 

 

정렬로 푼 코드

1
2
3
4
5
6
7
8
9
10
11
def solution(A, B):
    answer = 0
 
    A = sorted(A)
    B = sorted(B)
 
    for i in range(len(A)):
        answer += A[i] * B[len(A) - 1 - i]
    
    return answer
 
cs

 

 

 

바보같이 푼 코드(순열, 완전탐색) - 시간초과뜸

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
from itertools import permutations
 
 
def solution(A, B):
    answer = 123456789
 
    per_li = list(permutations(range(len(B)), len(B)))
 
    for permu in per_li:
        mul = 0
        for i in range(len(permu)):
            mul += A[i] * B[permu[i]]
        if answer > mul:
            answer = mul
 
    return answer
cs
728x90