示例#1
0
    """
    nums = [int(i) for i in lines]
    return solve_part1(nums, preamble)


def search(nums, target):
    for start in range(len(nums)):
        for length in range(len(nums)):
            my_slice = nums[start:length]
            if sum(my_slice) == target:
                return my_slice
    raise Exception


def part2(lines, preamble=25):
    """
    >>> part2(load_example(__file__, '9'), 5)
    62
    """
    nums = [int(i) for i in lines]
    target = solve_part1(nums, preamble)

    result = search(nums, target)
    return min(result) + max(result)


if __name__ == "__main__":
    data = load_input(__file__, 2020, "9")
    print(part1(data))
    print(part2(data))
示例#2
0
    grid, mxy = prepare_map(lines)
    cx = cy = mxy
    direction = 0
    infection_counter = 0
    for _ in range(n):
        current = grid[cx, cy]
        if current == CLEAN:
            grid[cx, cy] = WEAKENED
            direction = (direction + 3) % 4
        elif current == WEAKENED:
            grid[cx, cy] = INFECTED
            infection_counter += 1
        elif current == INFECTED:
            grid[cx, cy] = FLAGGED
            direction = (direction + 1) % 4
        elif current == FLAGGED:
            grid[cx, cy] = CLEAN
            direction = (direction + 2) % 4
        else:
            raise
        dx, dy = MOVEMENTS[direction]
        cx += dx
        cy += dy
    return infection_counter


if __name__ == "__main__":
    data = load_input(__file__, 2017, "22")
    print(part1(data))
    print(part2(data))
示例#3
0
def calc_happiness(happiness, arrangement):
    return happiness[(arrangement[0], arrangement[-1])] + sum(
        happiness[(arrangement[i - 1], arrangement[i])]
        for i in range(1, len(arrangement)))


def run(lines, with_me):
    """
    >>> run(load_example(__file__, '13'), False)
    330
    """
    guests, happiness = prepare_data(lines, with_me)
    return max(
        calc_happiness(happiness, arrangement)
        for arrangement in permutations(guests))


def part1(lines):
    return run(lines, with_me=False)


def part2(lines):
    return run(lines, with_me=True)


if __name__ == "__main__":
    data = load_input(__file__, 2015, "13")
    print(part1(data))
    print(part2(data))
示例#4
0

def find_loop(commands, initial_standing):
    standing = list(initial_standing)
    cache = {}
    i = 0
    while True:
        s = "".join(standing)
        if s in cache:
            # works because cache[s] is 0
            return cache, i
        cache[s] = i
        standing = dance(commands, standing)
        i += 1


def part2(lines, initial_standing="abcdefghijklmnop", count_dances=10 ** 9):
    """
    >>> part2(load_example(__file__, "16"), "abcde", 10)
    'ceadb'
    """
    commands = lines[0].strip().split(",")
    cache, s = find_loop(commands, initial_standing)
    return list(cache.keys())[count_dances % s]


if __name__ == "__main__":
    data = load_input(__file__, 2017, "16")
    print(part1(data))
    print(part2(data))
示例#5
0
            count_three += 1
    return count_two * count_three


def count_diff(line1, line2):
    count = 0
    common_str = ""
    for i in range(len(line1)):
        if line1[i] != line2[i]:
            count += 1
        else:
            common_str += line1[i]
    return count, common_str


def part2(lines):
    """
    >>> part2(load_example(__file__, '2b'))
    'fgij'
    """
    for line1, line2 in combinations(lines, 2):
        count, common = count_diff(line1.strip(), line2.strip())
        if count == 1:
            return common


if __name__ == "__main__":
    data = load_input(__file__, 2018, "2")
    print(part1(data))
    print(part2(data))
示例#6
0
    """
    return 1 + ((y - 1) * y + (x - 1) * x) // 2 + x * (y - 1)


def calc_manual_numbers(n):
    """
    >>> calc_manual_numbers(1)
    20151125
    >>> calc_manual_numbers(2)
    31916031
    >>> calc_manual_numbers(3)
    18749137
    >>> calc_manual_numbers(21)
    33511524
    """
    return (BASE * pow(FACTOR, n - 1, MOD)) % MOD


def part1(lines):
    x, y = prepare_data(lines)
    return calc_manual_numbers(calc_number_on(x, y))


def part2(lines):
    pass


if __name__ == "__main__":
    data = load_input(__file__, 2015, "25")
    print(part1(data))