* items의 value가 높은것 부터 채워 넣어야 함 * 같은 value는 합침 * 합치고 sort하면 sort 시간이 감소함 * 혹은 최대 value 개수의 table을 만들고 value에 각 weight들을 합(이것 자체를 sorted array로 사용) W, N = 100, 2 # capacity, num items items = [ [90, 1], # weight, value [70, 2] ] weight_sums = collections.defaultdict(int) for w, v in items: weight_sums[v] += w weight_sums = sorted(weight_sums.items(), key=lambda p: p[0], reverse=True) tot = 0 for ..

g = collections.defaultdict(dict) for u, v in relations: g[u][v] = 1 g[v][u] = 1 visited = set() cnt = 0 for node in range(1, N + 1): val = weights[node - 1] if len(g[node].items()) == 0: cnt += 1 continue for v, w in g[node].items(): if val

N = 2 size = 2**N grid = [[0]*size for _ in range(size)] rows = len(grid) cols = len(grid[0]) score = 0 for y in range(rows): for x in range(cols): if ((y == 0 and x == 0) or (y == rows - 1 and x == cols - 1) or (y == rows - 1 and x == 0) or (y == 0 and x == cols - 1)): score += 2.25 elif (y == 0 or x == 0 or x == cols - 1 or y == rows - 1): score += 1.5 else: score += 1 print(int(score))

뭐가 문제인건지.. 기본 TC는 통과.....흠.. N log (N) 알고리즘으로 구현하지 말고 그냥 NM으로 구현할까? 귀찮... import sys import bisect limits = [] test = [] N, M = map(int, sys.stdin.readline().split()) for i in range(N): limits += tuple(map(int, sys.stdin.readline().split())), for i in range(M): test += tuple(map(int, sys.stdin.readline().split())), tbl = [(0, 0)] start = 1 for length, speed in limits: tbl += (start + length - 1,..

grid = [] N = int(sys.stdin.readline()) for _ in range(N): line = str(sys.stdin.readline()).strip() line = list(line) line = list(map(int, line)) grid += line, def dfs(y, x): if 0 == grid[y][x]: return 0 cnt = 1 grid[y][x] = 0 for oy, ox in [(1, 0), (-1, 0), (0, 1), (0, -1)]: ny, nx = oy + y, ox + x if not (0

brute force algorithm ... N, M, K = map(int, sys.stdin.readline().split()) lines = list(map(int, sys.stdin.readline().split())) def get_sum(lines, K): cur = 0 idx = 0 tot = 0 while K: inc = 0 while cur + lines[idx]