직사각형 별찍기
첫 코드 a, b = map(int, input().strip().split(' ')) for j in range(1, b+1): print(a*'*')
- 알고리즘/프래그래머스 lv.1
- · 2022. 10. 18.
첫 코드 def solution(n): ans = [] temp = str(n) for i in range(len(temp)): ans.append(int(temp[i])) answer = sum(ans) return answer 개선 코드 def solution(n): return sum([int(i) for i in str(n)]) List Comprehension을 사용하여 간결하게 개선해 보았습니다.
문제 해설 arr1과 arr2의 합을 구하여 보자. 두 개의 배열은 2차원 배열로 행렬로 이루어져 있다. 핵심 아이디어 List Comprehension을 사용하여 2차원 배열의 합의 알고리즘을 짜보자. tmp = [[j for j in range(2)] for i in range(3)] tmp [[0, 1], [0, 1], [0, 1]] 첫 코드 def solution(arr1, arr2): answer = [[arr1[i][j] + arr2[i][j] for j in range(len(arr1[0]))] for i in range(len(arr1))] return answer
첫 코드 a, b = map(int, input().strip().split(' ')) for j in range(1, b+1): print(a*'*')
문제 설명 배열 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합니다. 핵심 아이디어 리스트의 양옆의 요소가 같으면 하나로 합치고, 다르면 반환한다. 첫 코드 def solution(arr): answer = [] answer.append(arr[0]) for i in range(0, len(arr)): if a..
문제 설명 대문자와 소문자가 섞여있는 문자열 s가 주어집니다. s에 'p'의 개수와 'y'의 개수를 비교해 같으면 True, 다르면 False를 return 하는 solution를 완성하세요. 'p', 'y' 모두 하나도 없는 경우는 항상 True를 리턴합니다. 단, 개수를 비교할 때 대문자와 소문자는 구별하지 않습니다. 첫 코드 def solution(s): s = s.lower() if s.count('p') == s.count('y'): answer = True elif s.count('p') != s.count('y'): answer = False else: answer = True return answer 개선된 코드 def solution(s): s = s.lower() if s.count('..
첫 코드 def solution(n): x = [] for i in range(1, n+1): if n % i == 0: x.append(i) answer = sum(x) return answer 개선된 코드 아래는 List comprehension으로 간략하게 쓴 코드입니다. def solution(n): return sum([i for i in range(1, n+1) if n % i == 0])