cal ollama

import sys def solve() -> None: data = sys.stdin.read().strip().split() if not data: return it = iter(data) t = int(next(it)) out_lines = [] for _ in range(t): n = int(next(it)) m = int(next(it)) d = int(next(it)) # day of next birthday (month is the same) ages = [int(next(it)) for _ in range(n)] res = [] for a in ages: if a == m - 1 or a == m: res.append(str(a)) out_lines.append(' '.join(res)) sys.stdout.write('\n'.join(out_lines)) if __name__ == "__main__": solve()

Post a Comment