#479987 (PyPy3) No.960 マンハッタン距離3

提出ソース

結果

問題	No.960 マンハッタン距離3
ユーザー	maspy
提出日時	2020-05-10 06:14:44
言語	PyPy3 (7.3.15)
結果	AC
実行時間	450 ms / 2,000 ms
コード長	6,278 bytes
コンパイル時間	446 ms
コンパイル使用メモリ	82,424 KB
実行使用メモリ	110,948 KB
最終ジャッジ日時	2024-07-06 23:39:06
合計ジャッジ時間	29,517 ms
ジャッジサーバーID （参考情報）	judge4 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	44 ms 56,796 KB
testcase_01	AC	41 ms 56,136 KB
testcase_02	AC	43 ms 56,496 KB
testcase_03	AC	42 ms 56,552 KB
testcase_04	AC	42 ms 56,452 KB
testcase_05	AC	43 ms 55,808 KB
testcase_06	AC	42 ms 56,280 KB
testcase_07	AC	42 ms 56,488 KB
testcase_08	AC	43 ms 56,116 KB
testcase_09	AC	43 ms 55,752 KB
testcase_10	AC	44 ms 57,152 KB
testcase_11	AC	43 ms 56,904 KB
testcase_12	AC	42 ms 56,432 KB
testcase_13	AC	42 ms 56,040 KB
testcase_14	AC	43 ms 55,484 KB
testcase_15	AC	43 ms 56,804 KB
testcase_16	AC	43 ms 56,172 KB
testcase_17	AC	42 ms 56,012 KB
testcase_18	AC	42 ms 55,736 KB
testcase_19	AC	44 ms 57,056 KB
testcase_20	AC	52 ms 69,240 KB
testcase_21	AC	52 ms 67,424 KB
testcase_22	AC	43 ms 56,192 KB
testcase_23	AC	46 ms 57,032 KB
testcase_24	AC	45 ms 57,364 KB
testcase_25	AC	44 ms 56,636 KB
testcase_26	AC	44 ms 56,360 KB
testcase_27	AC	46 ms 62,848 KB
testcase_28	AC	46 ms 61,436 KB
testcase_29	AC	42 ms 56,612 KB
testcase_30	AC	48 ms 63,772 KB
testcase_31	AC	49 ms 64,964 KB
testcase_32	AC	41 ms 56,672 KB
testcase_33	AC	51 ms 66,276 KB
testcase_34	AC	49 ms 65,140 KB
testcase_35	AC	44 ms 56,136 KB
testcase_36	AC	51 ms 65,720 KB
testcase_37	AC	44 ms 57,244 KB
testcase_38	AC	47 ms 62,480 KB
testcase_39	AC	48 ms 63,296 KB
testcase_40	AC	44 ms 57,100 KB
testcase_41	AC	42 ms 56,576 KB
testcase_42	AC	44 ms 56,808 KB
testcase_43	AC	44 ms 55,676 KB
testcase_44	AC	44 ms 56,992 KB
testcase_45	AC	41 ms 56,264 KB
testcase_46	AC	47 ms 62,828 KB
testcase_47	AC	45 ms 56,840 KB
testcase_48	AC	49 ms 64,380 KB
testcase_49	AC	49 ms 65,500 KB
testcase_50	AC	51 ms 62,784 KB
testcase_51	AC	45 ms 57,116 KB
testcase_52	AC	43 ms 56,568 KB
testcase_53	AC	50 ms 64,252 KB
testcase_54	AC	43 ms 56,712 KB
testcase_55	AC	42 ms 56,584 KB
testcase_56	AC	43 ms 56,296 KB
testcase_57	AC	43 ms 57,604 KB
testcase_58	AC	45 ms 56,636 KB
testcase_59	AC	43 ms 56,900 KB
testcase_60	AC	51 ms 64,756 KB
testcase_61	AC	47 ms 62,532 KB
testcase_62	AC	48 ms 64,212 KB
testcase_63	AC	50 ms 64,772 KB
testcase_64	AC	44 ms 56,836 KB
testcase_65	AC	46 ms 55,928 KB
testcase_66	AC	45 ms 56,436 KB
testcase_67	AC	45 ms 57,324 KB
testcase_68	AC	42 ms 57,412 KB
testcase_69	AC	44 ms 57,676 KB
testcase_70	AC	44 ms 56,240 KB
testcase_71	AC	44 ms 57,136 KB
testcase_72	AC	50 ms 64,308 KB
testcase_73	AC	49 ms 63,316 KB
testcase_74	AC	45 ms 57,784 KB
testcase_75	AC	44 ms 57,404 KB
testcase_76	AC	45 ms 56,376 KB
testcase_77	AC	43 ms 57,752 KB
testcase_78	AC	42 ms 57,472 KB
testcase_79	AC	46 ms 56,720 KB
testcase_80	AC	43 ms 57,276 KB
testcase_81	AC	42 ms 55,800 KB
testcase_82	AC	42 ms 56,400 KB
testcase_83	AC	43 ms 56,044 KB
testcase_84	AC	42 ms 56,280 KB
testcase_85	AC	43 ms 56,904 KB
testcase_86	AC	47 ms 56,608 KB
testcase_87	AC	44 ms 56,300 KB
testcase_88	AC	44 ms 57,560 KB
testcase_89	AC	48 ms 62,468 KB
testcase_90	AC	45 ms 57,824 KB
testcase_91	AC	44 ms 56,772 KB
testcase_92	AC	43 ms 56,996 KB
testcase_93	AC	44 ms 56,556 KB
testcase_94	AC	47 ms 62,540 KB
testcase_95	AC	47 ms 63,612 KB
testcase_96	AC	47 ms 63,096 KB
testcase_97	AC	49 ms 63,772 KB
testcase_98	AC	47 ms 56,500 KB
testcase_99	AC	45 ms 57,176 KB
testcase_100	AC	49 ms 64,556 KB
testcase_101	AC	47 ms 63,312 KB
testcase_102	AC	47 ms 63,496 KB
testcase_103	AC	43 ms 55,736 KB
testcase_104	AC	44 ms 56,308 KB
testcase_105	AC	42 ms 57,236 KB
testcase_106	AC	43 ms 57,032 KB
testcase_107	AC	44 ms 58,316 KB
testcase_108	AC	44 ms 57,576 KB
testcase_109	AC	49 ms 63,344 KB
testcase_110	AC	47 ms 57,252 KB
testcase_111	AC	44 ms 57,136 KB
testcase_112	AC	43 ms 55,880 KB
testcase_113	AC	43 ms 56,612 KB
testcase_114	AC	47 ms 55,876 KB
testcase_115	AC	43 ms 56,172 KB
testcase_116	AC	47 ms 56,152 KB
testcase_117	AC	42 ms 56,616 KB
testcase_118	AC	44 ms 56,136 KB
testcase_119	AC	42 ms 55,888 KB
testcase_120	AC	44 ms 57,896 KB
testcase_121	AC	44 ms 57,576 KB
testcase_122	AC	44 ms 56,148 KB
testcase_123	AC	43 ms 56,296 KB
testcase_124	AC	42 ms 56,568 KB
testcase_125	AC	43 ms 55,676 KB
testcase_126	AC	43 ms 56,796 KB
testcase_127	AC	42 ms 55,668 KB
testcase_128	AC	183 ms 83,692 KB
testcase_129	AC	247 ms 101,292 KB
testcase_130	AC	205 ms 91,516 KB
testcase_131	AC	266 ms 103,548 KB
testcase_132	AC	276 ms 104,000 KB
testcase_133	AC	117 ms 77,292 KB
testcase_134	AC	89 ms 82,876 KB
testcase_135	AC	167 ms 84,468 KB
testcase_136	AC	180 ms 84,272 KB
testcase_137	AC	109 ms 91,204 KB
testcase_138	AC	244 ms 98,132 KB
testcase_139	AC	250 ms 99,932 KB
testcase_140	AC	152 ms 110,428 KB
testcase_141	AC	270 ms 104,036 KB
testcase_142	AC	180 ms 88,148 KB
testcase_143	AC	133 ms 97,432 KB
testcase_144	AC	450 ms 101,488 KB
testcase_145	AC	305 ms 80,120 KB
testcase_146	AC	147 ms 81,564 KB
testcase_147	AC	361 ms 97,604 KB
testcase_148	AC	143 ms 86,384 KB
testcase_149	AC	145 ms 95,776 KB
testcase_150	AC	170 ms 101,412 KB
testcase_151	AC	92 ms 77,136 KB
testcase_152	AC	132 ms 99,416 KB
testcase_153	AC	151 ms 110,696 KB
testcase_154	AC	168 ms 78,372 KB
testcase_155	AC	328 ms 99,080 KB
testcase_156	AC	238 ms 82,284 KB
testcase_157	AC	285 ms 99,452 KB
testcase_158	AC	266 ms 105,508 KB
testcase_159	AC	176 ms 78,784 KB
testcase_160	AC	160 ms 78,312 KB
testcase_161	AC	316 ms 101,028 KB
testcase_162	AC	148 ms 103,700 KB
testcase_163	AC	128 ms 91,620 KB
testcase_164	AC	180 ms 106,196 KB
testcase_165	AC	186 ms 100,360 KB
testcase_166	AC	103 ms 78,884 KB
testcase_167	AC	133 ms 91,396 KB
testcase_168	AC	294 ms 84,404 KB
testcase_169	AC	306 ms 86,288 KB
testcase_170	AC	279 ms 83,800 KB
testcase_171	AC	311 ms 91,516 KB
testcase_172	AC	113 ms 88,084 KB
testcase_173	AC	99 ms 76,760 KB
testcase_174	AC	198 ms 101,992 KB
testcase_175	AC	204 ms 110,948 KB
testcase_176	AC	164 ms 99,636 KB
testcase_177	AC	104 ms 77,488 KB
testcase_178	AC	175 ms 99,204 KB
testcase_179	AC	179 ms 101,520 KB
testcase_180	AC	193 ms 103,308 KB
testcase_181	AC	121 ms 81,880 KB
testcase_182	AC	134 ms 88,300 KB
testcase_183	AC	141 ms 96,592 KB
testcase_184	AC	124 ms 89,584 KB
testcase_185	AC	162 ms 98,596 KB
testcase_186	AC	181 ms 102,116 KB
testcase_187	AC	175 ms 97,672 KB
testcase_188	AC	43 ms 57,200 KB
testcase_189	AC	43 ms 55,804 KB
testcase_190	AC	42 ms 56,844 KB
testcase_191	AC	42 ms 56,040 KB
testcase_192	AC	42 ms 56,552 KB
testcase_193	AC	42 ms 55,808 KB
testcase_194	AC	42 ms 56,572 KB
testcase_195	AC	43 ms 56,584 KB
testcase_196	AC	275 ms 97,940 KB
testcase_197	AC	200 ms 80,156 KB
testcase_198	AC	243 ms 90,552 KB
testcase_199	AC	276 ms 99,464 KB
testcase_200	AC	226 ms 101,708 KB
testcase_201	AC	178 ms 90,204 KB
testcase_202	AC	284 ms 101,600 KB
testcase_203	AC	244 ms 88,072 KB
testcase_204	AC	341 ms 104,504 KB
testcase_205	AC	230 ms 89,868 KB
testcase_206	AC	221 ms 103,180 KB
testcase_207	AC	226 ms 99,040 KB
testcase_208	AC	261 ms 99,604 KB
testcase_209	AC	174 ms 82,472 KB
testcase_210	AC	137 ms 78,448 KB
testcase_211	AC	221 ms 84,708 KB
testcase_212	AC	157 ms 82,336 KB
testcase_213	AC	192 ms 90,204 KB
testcase_214	AC	335 ms 105,796 KB
testcase_215	AC	290 ms 101,816 KB
testcase_216	AC	306 ms 104,176 KB
testcase_217	AC	245 ms 91,896 KB
testcase_218	AC	247 ms 106,200 KB
testcase_219	AC	188 ms 91,820 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

import sys
from collections import namedtuple

read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines

INF = 10**10

Segment = namedtuple('Segment', ('x', 'y', 'dx', 'dy', 'n'))
Rectangle = namedtuple('Rectangle', ('x1', 'y1', 'x2', 'y2'))
Pt = namedtuple('Pt', ('x', 'y'))

W, H = map(int, readline().split())
N = int(readline())
m = map(int, read().split())
X,Y = zip(*zip(m,m))
x0, y0 = X[0], Y[0]

def solve_two(x0, y0, x, y):
    if x0 > x:
        x0, x = x, x0
        y0, y = y, y0
    dx = x - x0
    dy = y - y0
    if (dx + dy) % 2 != 0:
        return
    if dx == abs(dy):
        if dy > 0:
            yield Rectangle(-INF, y, x0, INF)
            yield Rectangle(x, -INF, INF, y0)
            if dx > 1:
                yield Segment(x0 + 1, y - 1, 1, -1, dx - 2)
        else:
            yield Rectangle(-INF, -INF, x0, y)
            yield Rectangle(x, y0, INF, INF)
            if dx > 1:
                yield Segment(x0 + 1, y + 1, 1, 1, dx - 2)
        return
    d = (dx + abs(dy)) // 2
    if dx == 0:
        y = (y + y0) // 2
        yield Segment(-INF, y, 1, 0, INF * 2)
    elif dy == 0:
        x = (x + x0) // 2
        yield Segment(x, -INF, 0, 1, INF * 2)
    elif dx < dy:
        yield Segment(x0,y0+d,-1,0,INF)
        yield Segment(x, y-d, 1, 0, INF)
        if dx > 1:
            yield Segment(x0+1,y0+d-1,1,-1,dx-2)
    elif 0 < dy < dx:
        yield Segment(x-d,y,0,1,INF)
        if dy > 1:
            yield Segment(x - d + 1, y - 1, 1, -1, dy - 2)
        yield Segment(x0 + d, y0, 0, -1, INF)
    elif 0 < dx < -dy:
        yield Segment(x0, y0-d, -1, 0, INF)
        if dx > 1:
            yield Segment(x0+1,y0-d+1,1,1,dx - 2)
        yield Segment(x, y + d, 1, 0, INF)
    elif 0 < -dy < dx:
        yield Segment(x0+d,y0, 0, 1, INF)
        if -dy > 1:
            yield Segment(x-d+1,y+1, 1, 1, -dy - 2)
        yield Segment(x-d,y,0,-1,INF)
    else:
        raise Exception

def modify(seg):
    x,y,dx,dy,n = seg
    if dx == -1:
        return Segment(x+dx*n,y+dy*n,-dx,-dy,n)
    if dx == 0 and dy == -1:
        return Segment(x,y-n,0,1,n)
    return seg

def ptpt(pt1, pt2):
    if pt1.x == pt2.x and pt1.y == pt2.y:
        return pt1
    else:
        return None

def segpt(seg, pt):
    seg = modify(seg)
    d = abs(seg.x - pt.x) + abs(seg.y - pt.y)
    k = d // (abs(seg.dx) + abs(seg.dy))
    if not (0 <= k <= seg.n):
        return None
    if seg.x + seg.dx * k == pt.x and seg.y + seg.dy * k == pt.y:
        return pt
    return None

def recpt(rec, pt):
    if (rec.x1 <= pt.x <= rec.x2) and (rec.y1 <= pt.y <= rec.y2):
        return pt
    return None

def recrec(rec1, rec2):
    x1 = max(rec1.x1, rec2.x1)
    x2 = min(rec1.x2, rec2.x2)
    y1 = max(rec1.y1, rec2.y1)
    y2 = min(rec1.y2, rec2.y2)
    if x1 <= x2 and y1 <= y2:
        return Rectangle(x1, y1, x2, y2)
    return None

def recseg(rec, seg):
    seg = modify(seg)
    if seg.dx == 0:
        if not (rec.x1 <= seg.x <= rec.x2):
            return None
        low_x, high_x = 0, INF * 2
    elif seg.dx == 1:
        low_x, high_x = rec.x1 - seg.x, rec.x2 - seg.x
    else:
        low_x, high_x = seg.x - rec.x2, seg.x - rec.x1
    if seg.dy == 0:
        if not (rec.y1 <= seg.y <= rec.y2):
            return None
        low_y, high_y = 0, INF * 2
    elif seg.dy == 1:
        low_y, high_y = rec.y1 - seg.y, rec.y2 - seg.y
    else:
        low_y, high_y = seg.y - rec.y2, seg.y - rec.y1
    low = max(low_x, low_y, 0)
    high = min(high_x, high_y, seg.n)
    if low > high:
        return None
    return Segment(seg.x + seg.dx * low, seg.y + seg.dy * low, seg.dx, seg.dy,
                   high - low)

def segseg(seg1, seg2):
    seg1 = modify(seg1)
    seg2 = modify(seg2)
    x1, y1, a1, b1 = seg1.x, seg1.y, seg1.dx, seg1.dy
    x2, y2, a2, b2 = seg2.x, seg2.y, seg2.dx, seg2.dy
    det = -a1 * b2 + a2 * b1
    if det != 0:
        x, y = x2 - x1, y2 - y1
        s = (-b2 * x + a2 * y) // det
        t = (-b1 * x + a1 * y) // det
        if not ((0 <= s <= seg1.n) and (0 <= t <= seg2.n)):
            return None
        x, y = x1 + a1 * s, y1 + b1 * s
        if (x != x2 + a2 * t) or (y != y2 + b2 * t):
            return None
        return Pt(x, y)
    x1, y1, a1, b1 = seg1.x, seg1.y, seg1.dx, seg1.dy
    x2, y2, a2, b2 = seg2.x, seg2.y, seg2.dx, seg2.dy
    assert a1 == a2 and b1 == b2
    a, b = a1, b1
    if b * x1 - a * y1 != b * x2 - a * y2:
        # 交わらない平行線
        return None
    # 直線として同じ
    if a == 1:
        low_x = max(x1, x2)
        high_x = min(x1 + seg1.n, x2 + seg2.n)
        if low_x > high_x:
            return None
        return Segment(low_x, y1 + b * (low_x - x1), a, b, high_x - low_x)
    assert a == 0 and b == 1
    low_y = max(y1, y2)
    high_y = min(y1 + seg1.n, y2 + seg2.n)
    if low_y > high_y:
        return None
    return Segment(x1, low_y, 0, 1, high_y - low_y)

def intersect_two(shape1, shape2):
    if isinstance(shape1, Rectangle):
        if isinstance(shape2, Rectangle):
            return recrec(shape1, shape2)
        if isinstance(shape2, Segment):
            return recseg(shape1, shape2)
        return recpt(shape1, shape2)
    if isinstance(shape1, Segment):
        if isinstance(shape2, Rectangle):
            return recseg(shape2, shape1)
        if isinstance(shape2, Segment):
            return segseg(shape1, shape2)
        return segpt(shape1, shape2)
    if isinstance(shape2, Rectangle):
        return recpt(shape2, shape1)
    if isinstance(shape2, Segment):
        return segpt(shape2, shape1)
    return ptpt(shape1, shape2)

def intersect(pts1, pts2):
    for shape1 in pts1:
        for shape2 in pts2:
            shape = intersect_two(shape1, shape2)
            if shape is None:
                continue
            yield shape

def count_pt(shape):
    if isinstance(shape, Rectangle):
        return (shape.x2 - shape.x1 + 1) * (shape.y2 - shape.y1 + 1)
    if isinstance(shape, Segment):
        return shape.n + 1
    return 1

pts = [Rectangle(1, 1, W, H)]
for x, y in zip(X[1:], Y[1:]):
    pts1 = list(solve_two(x0, y0, x, y))
    pts = list(intersect(pts, pts1))

answer = 0
for shape in pts:
    answer += count_pt(shape)

print(answer)

yukicoder

結果

テストケース

ソースコード