Algorithm

[ 알고리즘 ] 완전탐색 - 최소 직사각형, 모의고사 문제

jogaknabi_1023 2024. 4. 16. 23:08

최소 직사각형 문제)

명함 지갑을 만드는 회사에서 지갑의 크기를 정하려고 합니다. 다양한 모양과 크기의 명함들을 모두 수납할 수 있으면서, 작아서 들고 다니기 편한 지갑을 만들어야 합니다. 이러한 요건을 만족하는 지갑을 만들기 위해 디자인팀은 모든 명함의 가로 길이와 세로 길이를 조사했습니다.

 

가장 긴 가로 길이와 세로 길이가 각각 80, 70이기 때문에 80(가로) x 70(세로) 크기의 지갑을 만들면 모든 명함들을 수납할 수 있습니다. 하지만 2번 명함을 가로로 눕혀 수납한다면 80(가로) x 50(세로) 크기의 지갑으로 모든 명함들을 수납할 수 있습니다. 이때의 지갑 크기는 4000(=80 x 50)입니다.

모든 명함의 가로 길이와 세로 길이를 나타내는 2차원 배열 sizes가 매개변수로 주어집니다. 모든 명함을 수납할 수 있는 가장 작은 지갑을 만들 때, 지갑의 크기를 return 하도록 solution 함수를 완성해주세요.

 

 

코드)

def solution(sizes):
    # 각 명함을 검토하여 가로와 세로 중 큰 값을 가로로, 작은 값을 세로로 사용.
    # 이렇게 함으로써 모든 명함을 회전시켜 효율적인 배치가 가능하다.
    adjusted_sizes = [[max(w, h), min(w, h)] for w, h in sizes]
    
    # 조정된 명함들 중에서 가로 길이의 최댓값과 세로 길이의 최댓값을 찾는다.
    max_width = max(size[0] for size in adjusted_sizes)
    max_height = max(size[1] for size in adjusted_sizes)
    
    # 최대 가로 길이와 최대 세로 길이를 곱하여 지갑의 최소 크기를 반환한다.
    return max_width * max_height
  • sizes 배열의 각 명함에 대해 [더 긴 길이, 더 짧은 길이] 형태로 새로운 배열을 생성한다.
  • max(w, h): 이 부분은 w와 h 중 더 큰 값을 반환하고, min(w, h): 이 부분은 w와 h 중 더 작은 값을 반환한다.

 

 

 

모의고사 문제)

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 1번 문제부터 마지막 문제까지 다음과 같이 찍습니다.

1번 수포자가 찍는 방식: 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, ...
2번 수포자가 찍는 방식: 2, 1, 2, 3, 2, 4, 2, 5, 2, 1, 2, 3, 2, 4, 2, 5, ...
3번 수포자가 찍는 방식: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, ...

1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이 누구인지 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한 조건
시험은 최대 10,000 문제로 구성되어있습니다.
문제의 정답은 1, 2, 3, 4, 5중 하나입니다.
가장 높은 점수를 받은 사람이 여럿일 경우, return하는 값을 오름차순 정렬해주세요.

 

코드)

def solution(answers):
    patterns = [
        [1, 2, 3, 4, 5],
        [2, 1, 2, 3, 2, 4, 2, 5],
        [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    ]
    scores = [0, 0, 0]

    for idx in range(len(answers)):
        answer = answers[idx]
        for j in range(3):
            if answer == patterns[j][idx % len(patterns[j])]:
                scores[j] += 1

    max_score = max(scores)
    result = []
    for i in range(3):
        if scores[i] == max_score:
            result.append(i + 1)

    return result

 

 

  • j 를 통해 몇 번째 수포자인지 구별한다.
  • if answer == patterns[j][idx % len(patterns[j])] -> 각각의 answers[idx]에 patterns[j]의 길이를 나눈 나머지를 통해 patterns이 answer의 요소보다 많더라도 patterns의 요소들을 한 번씩 사용한 후 다시 반복적으로 patterns[j]의 처음 요소와 answer 요소와의 비교를 시작한다.
  • 마지막으로 result 배열에 가장 높은 점수인 max_score 과 같은 사람을 추가한다.