Python - 함수 - 백준 알고리즘
Updated:
백준 알고리즘 함수 파이썬 풀이입니다.
15596번 정수 N개의 합
문제
정수 n개가 주어졌을 때, n개의 합을 구하는 함수를 작성하시오.
작성해야 하는 함수는 다음과 같다.
C, C11, C (Clang), C11 (Clang): long long sum(int *a, int n);
a: 합을 구해야 하는 정수 n개가 저장되어 있는 배열 (0 ≤ a[i] ≤ 1,000,000, 1 ≤ n ≤ 3,000,000)
n: 합을 구해야 하는 정수의 개수
리턴값: a에 포함되어 있는 정수 n개의 합
C++, C++11, C++14, C++17, C++ (Clang), C++11 (Clang), C++14 (Clang), C++17 (Clang): long long sum(std::vector<int> &a);
a: 합을 구해야 하는 정수 n개가 저장되어 있는 배열 (0 ≤ a[i] ≤ 1,000,000, 1 ≤ n ≤ 3,000,000)
리턴값: a에 포함되어 있는 정수 n개의 합
Python 2, Python 3, PyPy, PyPy3: def solve(a: list) -> int
a: 합을 구해야 하는 정수 n개가 저장되어 있는 리스트 (0 ≤ a[i] ≤ 1,000,000, 1 ≤ n ≤ 3,000,000)
리턴값: a에 포함되어 있는 정수 n개의 합 (정수)
Java: long sum(int[] a); (클래스 이름: Test)
a: 합을 구해야 하는 정수 n개가 저장되어 있는 배열 (0 ≤ a[i] ≤ 1,000,000, 1 ≤ n ≤ 3,000,000)
리턴값: a에 포함되어 있는 정수 n개의 합
Go: sum(a []int) int
a: 합을 구해야 하는 정수 n개가 저장되어 있는 배열 (0 ≤ a[i] ≤ 1,000,000, 1 ≤ n ≤ 3,000,000)
리턴값: a에 포함되어 있는 정수 n개의 합
저는 파이썬으로 해결하였습니다.
정답
def solve(a):
ans = 0
for i in a:
ans = ans + i
return ans
4673번 셀프 넘버
문제
N셀프 넘버는 1949년 인도 수학자 D.R. Kaprekar가 이름 붙였다. 양의 정수 n에 대해서 d(n)을 n과 n의 각 자리수를 더하는 함수라고 정의하자.
예를 들어, d(75) = 75+7+5 = 87이다. 양의 정수 n이 주어졌을 때, 이 수를 시작해서 n, d(n), d(d(n)), d(d(d(n))), …과 같은 무한 수열을 만들 수 있다.
예를 들어, 33으로 시작한다면 다음 수는 33 + 3 + 3 = 39이고, 그 다음 수는 39 + 3 + 9 = 51, 다음 수는 51 + 5 + 1 = 57이다. 이런식으로 다음과 같은 수열을 만들 수 있다.
33, 39, 51, 57, 69, 84, 96, 111, 114, 120, 123, 129, 141, …
n을 d(n)의 생성자라고 한다. 위의 수열에서 33은 39의 생성자이고, 39는 51의 생성자, 51은 57의 생성자이다. 생성자가 한 개보다 많은 경우도 있다. 예를 들어, 101은 생성자가 2개(91과 100) 있다.
생성자가 없는 숫자를 셀프 넘버라고 한다. 100보다 작은 셀프 넘버는 총 13개가 있다.
1, 3, 5, 7, 9, 20, 31, 42, 53, 64, 75, 86, 97
10000보다 작거나 같은 셀프 넘버를 한 줄에 하나씩 출력하는 프로그램을 작성하시오.
입력
입력은 없다.
출력
10,000보다 작거나 같은 셀프 넘버를 한 줄에 하나씩 증가하는 순서로 출력한다.
1
3
5
7
9
20
31
42
53
64
|
| <-- a lot more numbers
|
9903
9914
9925
9927
9938
9949
9960
9971
9982
9993
정답
def find_self_number(ans):
i = 1
while (i < 10000): # 10000까지 돌려야하므로
sum = i # 원래 수도 더해야하니까 설정
x = i # 계속 나누는걸 업데이트 하기 위해 x로 따로 설정
while True: # 각 수에 대해 d(n)을 구해서 1로 바꾼다.
quotient = x // 10 # 몫을 구함, 다시 나누기 위해
remainder = x % 10 # 나머지를 구함 , 더하기 위해
sum = sum + remainder # 나머지을 sum에 더해
x = quotient # x를 나머지로 바꾸고 다시 얘를 나눌거임
if (quotient == 0): # 더이상 나눌 수 없을 땐 한자리만 남은거니까 while 탈출
break
if sum < 10000: # d(n)의 결과물인 sum이 10000보다 작으면 원하는 값 맞으니까 1로 바꾼다.
ans[sum] = 1
i = i + 1
for i in range(1, len(ans)):
if ans[i] == 0:
print(i)
ans = [0 for i in range(10000)] # 1만개 모두 0으로 선언, 셀프 넘버가 아니게 되면 1로 바꿀거고 다 끝나며 0인것만 출력할 것
find_self_number(ans)
1065번 한수
문제
어떤 양의 정수 X의 각 자리가 등차수열을 이룬다면, 그 수를 한수라고 한다. 등차수열은 연속된 두 개의 수의 차이가 일정한 수열을 말한다. N이 주어졌을 때, 1보다 크거나 같고, N보다 작거나 같은 한수의 개수를 출력하는 프로그램을 작성하시오.
입력
첫째 줄에 1,000보다 작거나 같은 자연수 N이 주어진다.
210
출력
첫째 줄에 1보다 크거나 같고, N보다 작거나 같은 한수의 개수를 출력한다.
105
정답
N = int(input()) # 입력받을 수
# remember = [] # 하나씩 쪼개서 저장할 리스트
def COUNT(N):
remember = [] # 하나씩 쪼개서 저장할 리스트
count = 0
i = 1
while i <= N:
# 한 자리수 씩 쪼개서 저장
remember.clear() # remember 초기화
product = i # 입력한 숫자 저장
while True:
quotient = product // 10 # 몫 저장
remainder = product % 10 # 나머지 저장
remember.append(remainder) # 한 자리씩 remember에 추가
product = quotient
if product == 0:
break
TF = True
# rememeber에 한 자리 수 씩 저장 완료
# 이제 각 자리의 차이가 일정한지 비교하고 일정하면 count를 추가
if len(remember) == 1 or len(remember) == 2: # N이 99까지는 무조건 1 count
count = count + 1
else: # 100부터는 계산해봐야 함
D = remember[0] - remember[1] # 첫 수와 두 번째 수의 차이를 일단 D로 정의하고 비교할 것임
for j in range(1, len(remember) - 1):
if remember[j] - remember[j + 1] != D: # 하나라도 차가 다르면 TF를 False로 바꾸고 탈출
TF = False
break
if TF: # TF가 True이면 등차수열인 것이므로 count 증가
count = count + 1
# i를 1씩 증가시키면서 반복
i = i + 1
return count
print(COUNT(N))
Leave a comment