흙금이네 블로그

아이디어 N의 자릿값에 따라 각 숫자가 나오는 횟수를 계산한다. 풀이 def solution(): N = int(input()) number = list(map(int, [*str(N)]))[::-1] M = len(number) res = [0]*10 for i in range(M-1): p = 10**i if i == 0: res[0] += N//10 else: if number[i] > 0: res[0] += (N//(p*10))*p else: res[0] += (N//(p*10)-1)*p+(N%p+1) for i in range(M): p = 10**i for num in range(1, 10): if number[i] > num: if i == 0: res[num] += N//10+1 elif ..

아이디어 동적 계획법으로 서로 인접하지 않은 색을 고를 수 있는 경우의 수를 구한다. 풀이 def solution(): N = int(input()) K = int(input()) p = 1000000003 if K > N//2: print(0) return dp = [[0]*N for _ in range(N//2+1)] dp[1] = [1]*N for i in range(2, N//2+1): for j in range(N-3): dp[i][j+2] = (dp[i][j+1]+dp[i-1][j])%p dp[i][-1] = dp[i][-2] res = 0 for j in range(N): res = (res+dp[K][j])%p print(res) solution()

아이디어 #1 BFS로 세 그룹의 돌이 개수가 같아질 수 있는지 확인한다. 풀이 #1 import sys input = sys.stdin.readline def solution(): A, B, C = map(int, input().split()) if (A+B+C)%3: print(0) else: visited = dict() queue = [tuple(sorted([A, B, C]))] visited[queue[0]] = 1 while queue: a, b, c = queue.pop(0) if a == b == c: print(1) break if a != b: temp = tuple(sorted([a+a, b-a, c])) if visited.get(temp, 0) == 0: visited[temp]..