글 읽기 - 간이 채점기를 짰습니다

sait2000 1년 전 1

-----------------------------------------------

글이 너무 짧으면 등록이 안 되더라고요

-----------------------------------------------

import sys

TYPE_REG = 'REG'
TYPE_SHIFT = 'SHIFT'
OP_INFO = [
    ('SET', TYPE_REG, TYPE_REG),
    ('XOR', TYPE_REG, TYPE_REG, TYPE_REG),
    ('AND', TYPE_REG, TYPE_REG, TYPE_REG),
    ('OR', TYPE_REG, TYPE_REG, TYPE_REG),
    ('NOT', TYPE_REG, TYPE_REG),
    ('LSH', TYPE_REG, TYPE_SHIFT),
    ('RSH', TYPE_REG, TYPE_SHIFT),
]

ops = []

for line in iter(sys.stdin.readline, ''):
    op_type, *args = line.split()
    success = False
    for op_type_known, *args_known in OP_INFO:
        if op_type != op_type_known:
            continue
        if len(args) != len(args_known):
            break
        success = True
        args_converted = []
        for arg_given, arg_known in zip(args, args_known):
            try:
                arg_given = int(arg_given)
            except ValueError:
                success = False
                break
            lo, hi = (1, 400) if arg_known == TYPE_REG else (0, 63)
            if not (lo <= arg_given <= hi):
                success = False
                break
            args_converted.append(arg_given)
        if success:
            ops.append((op_type, args_converted))
        break
    if not success:
        print('Invalid Operation:', line.rstrip('\n'), file=sys.stderr)
        sys.exit(1)


import random

success = True

ttt = 0

for __ in range(100):
    regs = [0] * 401
    t = [0] * 401
    tt = 0

    x = random.randrange(2**64)
    y = random.randrange(2**64)
    res = (x * y) % (2**64)
    regs[1] = x
    regs[2] = y
    for op_type, args in ops:
        if op_type == 'SET':
            i, j = args
            regs[i] = regs[j]
            t[i] = t[j] + 1
            tt = max(tt, t[i])
        if op_type == 'XOR':
            i, j, k = args
            regs[i] = regs[j] ^ regs[k]
            t[i] = max(t[j], t[k]) + 1
            tt = max(tt, t[i])
        if op_type == 'AND':
            i, j, k = args
            regs[i] = regs[j] & regs[k]
            t[i] = max(t[j], t[k]) + 1
            tt = max(tt, t[i])
        if op_type == 'OR':
            i, j, k = args
            regs[i] = regs[j] | regs[k]
            t[i] = max(t[j], t[k]) + 1
            tt = max(tt, t[i])
        if op_type == 'NOT':
            i, j = args
            regs[i] = regs[j] ^ (2**64 - 1)
            t[i] = t[j] + 1
            tt = max(tt, t[i])
        if op_type == 'LSH':
            i, x = args
            regs[i] = (regs[i] << x) & (2**64 - 1)
            t[i] = t[i] + 1
            tt = max(tt, t[i])
        if op_type == 'RSH':
            i, x = args
            regs[i] = (regs[i] >> x) & (2**64 - 1)
            t[i] = t[i] + 1
            tt = max(tt, t[i])
    ttt = max(ttt, tt)
    if regs[1] != res:
        print(f'Wrong Answer: {x} * {y} = {res} != {regs[1]}', file=sys.stderr)
        sys.exit(1)
    if ttt > 70:
        print('TLE')
        sys.exit(1)

print(ttt)

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

간이 채점기를 짰습니다

sait2000 1년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말