TIL27. algorithm 기초

2016년

문제 설명
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일같은 날짜는 주어지지 않습니다)

입출력 예
a b result
5 24 "TUE"

import datetime
def solution(a,b):
    answer = ['MON', 'TUE', 'WED', 'THU', 'FRI', 'SAT', 'SUN']
    num = datetime.date(2016,a,b).weekday()
    return answer[num]
******** datetime.date().weekday() 는  월(0)~일(6) 까지 표시해주는 함수 ********

가운데 글자 가져오기

문제 설명
단어 s의 가운데 글자를 반환하는 함수, solution을 만들어 보세요. 단어의 길이가 짝수라면 가운데 두글자를 반환하면 됩니다.

재한사항
s는 길이가 1 이상, 100이하인 스트링입니다.

입출력 예
"abcde" "c"
"qwer" "we"

def solution(s):
  if len(s) % 2 == 1:
    answer = (s[len(s) // 2])
  else:
    answer = (s[len(s) // 2 - 1] + s[len(s) // 2])
  return answer

같은 숫자는 싫어

문제 설명
배열 arr가 주어집니다. 배열 arr의 각 원소는 숫자 0부터 9까지로 이루어져 있습니다. 이때, 배열 arr에서 연속적으로 나타나는 숫자는 하나만 남기고 전부 제거하려고 합니다. 단, 제거된 후 남은 수들을 반환할 때는 배열 arr의 원소들의 순서를 유지해야 합니다. 예를 들면,

arr = [1, 1, 3, 3, 0, 1, 1] 이면 [1, 3, 0, 1] 을 return 합니다.
arr = [4, 4, 4, 3, 3] 이면 [4, 3] 을 return 합니다.
배열 arr에서 연속적으로 나타나는 숫자는 제거하고 남은 수들을 return 하는 solution 함수를 완성해 주세요.

제한사항
배열 arr의 크기 : 1,000,000 이하의 자연수
배열 arr의 원소의 크기 : 0보다 크거나 같고 9보다 작거나 같은 정수

입출력 예

[1,1,3,3,0,1,1][1,3,0,1]
[4,4,4,3,3][4,3]

def solution(arr): 
    answer = [] 					# 답이 될 빈 배열을 만들어준다.
    count = 0 
    for i in arr: 					# arr 안에 있는 값들을 돌아준다.
        if len(answer) == 0:				# 처음에는 answer 안에 아무 값도 없으므로  
            answer.append(i) 				# 값을 대입해주고 
        if i == answer[count]: 				# answer안에 들어있는 값과 arr 의 값이 같으면  
            continue 					# answer에 넣으면 안되니까 continue 해주고 
        else: 						# arr 배열 안의 값이 answer에 있는 값과 다르면 
            answer.append(i) 				# 그 값을 answer 리스트안에 넣어주고
            count += 1 					# answer[count]가 방금 넣은 값을 가르키도록 count 를 1 올려준다.       
    return answer

나누어 떨어지는 숫자 배열

문제 설명
array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요.
divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하세요.

제한사항
arr은 자연수를 담은 배열입니다.
정수 i, j에 대해 i ≠ j 이면 arr[i] ≠ arr[j] 입니다.
divisor는 자연수입니다.
array는 길이 1 이상인 배열입니다.

입출력 예
[5, 9, 7, 10] 5 [5, 10]

[2, 36, 1, 3] 1 [1, 2, 3, 36]

[3,2,6] 10 [-1]

def solution(arr, divisor):
    answer = []						# 출력할 답 공백 리스트
    for i in range(len(arr)):				# 입력 받은 배열 길이만큼 for문 실행
        if arr[i] % divisor == 0:			# 그 배열 안에 있는 숫자가 입력받은 수와 딱 맞게 떨어진다면
            answer.append(arr[i])			# answer 리스트에 추가
    if len(answer) == 0:				# 만약 answer 리스트가 0이라면
        answer = [-1]					# answer는 -1 이다.
    return sorted(answer)				# 정렬된 answer 리스트를 출력한다.

두 정수 사이의 합

문제 설명
두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요.
예를 들어 a = 3, b = 5인 경우, 3 + 4 + 5 = 12이므로 12를 리턴합니다.

제한 조건
a와 b가 같은 경우는 둘 중 아무 수나 리턴하세요.
a와 b는 -10,000,000 이상 10,000,000 이하인 정수입니다.
a와 b의 대소관계는 정해져있지 않습니다.

입출력 예
a b return
3 5 12
3 3 3
5 3 12

  def solution(a, b):
    answer = 0					# answer를 0으로 선언
    for i in range(min(a,b),max(a,b)+1):	# a,b 두개사이의 길이만큼 for문 생성
        answer += i				# answer에 그대로 추가
    return answer

문자열 내 마음 대로 정렬하기

문제 설명
문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱스 1의 문자 "u", "e", "a"로 strings를 정렬합니다.

제한 조건
strings는 길이 1 이상, 50이하인 배열입니다.
strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
모든 strings의 원소의 길이는 n보다 큽니다.
인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

입출력 예
strings n return
["sun", "bed", "car"] 1 ["car", "bed", "sun"]

["abce", "abcd", "cdx"] 2 ["abcd", "abce", "cdx"]

  def solution(strings, n):
    strings.sort()
    strings.sort(key=lambda x:x[n])
    return strings

구글검색

문자열 내 p와 y의 개수

문제 설명
대문자와 소문자가 섞여있는 문자열 s가 주어집니다. s에 'p'의 개수와 'y'의 개수를 비교해 같으면 True, 다르면 False를 return 하는 solution를 완성하세요. 'p', 'y' 모두 하나도 없는 경우는 항상 True를 리턴합니다. 단, 개수를 비교할 때 대문자와 소문자는 구별하지 않습니다.

예를 들어 s가 "pPoooyY"면 true를 return하고 "Pyy"라면 false를 return합니다.

제한사항
문자열 s의 길이 : 50 이하의 자연수
문자열 s는 알파벳으로만 이루어져 있습니다.

입출력 예
s answer
"pPoooyY" true
"Pyy" false

def solution(s):    	
    s = list(s)								# 입력받은 s를 리스트로 변환
    if (s.count('p') + s.count('P')) == (s.count('y') + s.count('Y')):	# 대소문자 구분없이 p와 y 갯수가 같다면
        return True							# True 출력
    else:								# 다르다면
        return False							# False 출력

문자열 내림차순으로 배열하기

문제 설명
문자열 s에 나타나는 문자를 큰것부터 작은 순으로 정렬해 새로운 문자열을 리턴하는 함수, solution을 완성해주세요.
s는 영문 대소문자로만 구성되어 있으며, 대문자는 소문자보다 작은 것으로 간주합니다.

제한 사항
str은 길이 1 이상인 문자열입니다.

입출력 예
"Zbcdefg" "gfedcbZ"

 def solution(s):			
    s = list(s)				#입력받은 문자열s를 리스트로 변환
    s.sort(reverse = True)		# 내림차순으로 정렬
    return (''.join(s))			# 리스트로 나눠놓은 것을 이어 붙힘

문자열 다루기 기본

문제 설명
문자열 s의 길이가 4 혹은 6이고, 숫자로만 구성돼있는지 확인해주는 함수, solution을 완성하세요. 예를 들어 s가 "a234"이면 False를 리턴하고 "1234"라면 True를 리턴하면 됩니다.

제한 사항
s는 길이 1 이상, 길이 8 이하인 문자열입니다.

입출력 예
"a234" false
"1234" true

 def solution(s):
    return (len(s) == 4 or len(s) == 6) and s.isdigit()	 # 문자길이가 4혹은 6 그리고 전부 숫자면 True 출력

소수 찾기

문제 설명
1부터 입력받은 숫자 n 사이에 있는 소수의 개수를 반환하는 함수, solution을 만들어 보세요.

소수는 1과 자기 자신으로만 나누어지는 수를 의미합니다.
(1은 소수가 아닙니다.)

제한 조건
n은 2이상 1000000이하의 자연수입니다.

입출력 예
10 4
5 3

def solution(n):
    num= set(range(2,n+1))
    for i in range(2,n+1):
        if i in num:
            num -=set(range(2*i, n+1, i))
    return len(num)