20240116/ programmers/ 1단계/ python

1. 코딩테스트 연습 - 2016년 | 프로그래머스 스쿨 (programmers.co.kr)

문제 설명

2016년 1월 1일은 금요일입니다. 2016년 a월 b일은 무슨 요일일까요? 두 수 a ,b를 입력받아 2016년 a월 b일이 무슨 요일인지 리턴하는 함수, solution을 완성하세요. 요일의 이름은 일요일부터 토요일까지 각각 SUN,MON,TUE,WED,THU,FRI,SAT

입니다. 예를 들어 a=5, b=24라면 5월 24일은 화요일이므로 문자열 "TUE"를 반환하세요.

 

제한 조건

• 2016년은 윤년입니다.
• 2016년 a월 b일은 실제로 있는 날입니다. (13월 26일이나 2월 45일같은 날짜는 주어지지 않습니다)

나의 풀이

.

모범 답안

def getDayName(a,b):
    month = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    day = ['FRI', 'SAT', 'SUN', 'MON', 'TUE', 'WED', 'THU']
    return day[(sum(month[:a-1])+b-1)%7]

 

2. 코딩테스트 연습 - 폰켓몬 | 프로그래머스 스쿨 (programmers.co.kr)

3. 코딩테스트 연습 - 과일 장수 | 프로그래머스 스쿨 (programmers.co.kr)

4. 코딩테스트 연습 - 소수 만들기 | 프로그래머스 스쿨 (programmers.co.kr)

문제 설명

주어진 숫자 중 3개의 수를 더했을 때 소수가 되는 경우의 개수를 구하려고 합니다. 숫자들이 들어있는 배열 nums가 매개변수로 주어질 때, nums에 있는 숫자들 중 서로 다른 3개를 골라 더했을 때 소수가 되는 경우의 개수를 return 하도록 solution 함수를 완성해주세요.

 

 

제한사항

• nums에 들어있는 숫자의 개수는 3개 이상 50개 이하입니다.
• nums의 각 원소는 1 이상 1,000 이하의 자연수이며, 중복된 숫자가 들어있지 않습니다.

나의 풀이

.

 

모범 답안

def solution(nums):
    from itertools import combinations as cb
    answer = 0
    for a in cb(nums, 3):
        cand = sum(a)
        for j in range(2, cand):
            if cand%j==0:
                break
        else:
            answer += 1
    return answer

 ★ 조합 (Combination) : combinations(배열, 뽑는 개수)

    -  itertools 패키지에서 combinations를 이용하여 순열을 구할 수 있음

 

5. 코딩테스트 연습 - [1차] 비밀지도 | 프로그래머스 스쿨 (programmers.co.kr)

문제 설명 

네오는 평소 프로도가 비상금을 숨겨놓는 장소를 알려줄 비밀지도를 손에 넣었다. 그런데 이 비밀지도는 숫자로 암호화되어 있어 위치를 확인하기 위해서는 암호를 해독해야 한다. 다행히 지도 암호를 해독할 방법을 적어놓은 메모도 함께 발견했다.

1. 지도는 한 변의 길이가 n인 정사각형 배열 형태로, 각 칸은 "공백"(" ") 또는 "벽"("#") 두 종류로 이루어져 있다.
2. 전체 지도는 두 장의 지도를 겹쳐서 얻을 수 있다. 각각 "지도 1"과 "지도 2"라고 하자. 지도 1 또는 지도 2 중 어느 하나라도 벽인 부분은 전체 지도에서도 벽이다. 지도 1과 지도 2에서 모두 공백인 부분은 전체 지도에서도 공백이다.
3. "지도 1"과 "지도 2"는 각각 정수 배열로 암호화되어 있다.
4. 암호화된 배열은 지도의 각 가로줄에서 벽 부분을 1, 공백 부분을 0으로 부호화했을 때 얻어지는 이진수에 해당하는 값의 배열이다.

네오가 프로도의 비상금을 손에 넣을 수 있도록, 비밀지도의 암호를 해독하는 작업을 도와줄 프로그램을 작성하라.

 

입력 형식 

입력으로 지도의 한 변 크기 n 과 2개의 정수 배열 arr1, arr2가 들어온다.

  · 1 ≦ n ≦ 16

  ·  arr1, arr2는 길이 n인 정수 배열로 주어진다. 

  ·  정수 배열의 각 원소 x를 이진수로 변환했을 때의 길이는 n 이하이다.  즉, 0 ≦ x ≦ 2n - 1을 만족한다.

 

출력 형식

원래의 비밀지도를 해독하여 '#', 공백으로 구성된 문자열 배열로 출력하라.

 

나의 풀이 (실패)

def solution(n, arr1, arr2):
    ar1=[]
    ar2=[]
    result=[]
    string=""
    for i1 in arr1:
        ar1.append(int(bin(i1)[2:]))
    for i2 in arr2:
        ar2.append(int(bin(i2)[2:]))
    arr=dict(zip(ar1, ar2))
    
    for i in arr.items():
        num=str(int(i[0])+int(i[1]))
        if len(num)<n:
            num="0"*(n-len(num))+num
        for j in num:
            if j=="0":
                string+=" "
            else:
                string+="#"
                
        result.append(string) 
    return result

 

모범 답안

def solution(n, arr1, arr2):
    answer = []
    for i in range(n):
        a = str(bin(arr1[i]|arr2[i])[2:]).rjust(n,'0').replace('1','#').replace('0',' ')
        answer.append(a)
    return answer

def solution(n, arr1, arr2):
    answer = []
    for i in range(n):
        a12 = str(bin(arr1[i]|arr2[i])[2:])
        a12=a12.rjust(n,'0')
        a12=a12.replace('1','#')
        a12=a12.replace('0',' ')
        answer.append(a12)
    return answer

▶ bin(arr1[i] | arr2[i])는 각각의 지도를 이진수로 변환해서 OR 연산을 한 것.

     두 지도 모두 공백일 때(0일 때)만 공백이므로 OR 연산을 수행.

 

★ rjust

  rjust 함수는 문자열을 오른쪽 정렬해서 보기좋게 출력할 때 쓴다.

  형태는  문자열.rjust(전체 자리 숫자, 공백이 있을 경우 공백을 채울 텍스트) 

파이썬) rjust 함수 사용법 (오른쪽 정렬해서 출력하기) (tistory.com)