#936987 (Nim) No.2606 Mirror Relay

提出ソース

結果

問題	No.2606 Mirror Relay
ユーザー	👑 seekworser
提出日時	2023-12-07 02:41:22
言語	Nim (2.0.2)
結果	AC
実行時間	288 ms / 2,000 ms
コード長	11,198 bytes
コンパイル時間	5,687 ms
コンパイル使用メモリ	91,656 KB
実行使用メモリ	85,740 KB
最終ジャッジ日時	2024-01-02 13:43:52
合計ジャッジ時間	9,595 ms
ジャッジサーバーID （参考情報）	judge11 / judge14

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 6,548 KB
testcase_01	AC	1 ms 6,548 KB
testcase_02	AC	1 ms 6,548 KB
testcase_03	AC	1 ms 6,548 KB
testcase_04	AC	2 ms 6,548 KB
testcase_05	AC	2 ms 6,548 KB
testcase_06	AC	2 ms 6,548 KB
testcase_07	AC	2 ms 6,548 KB
testcase_08	AC	2 ms 6,548 KB
testcase_09	AC	2 ms 6,548 KB
testcase_10	AC	2 ms 6,548 KB
testcase_11	AC	2 ms 6,548 KB
testcase_12	AC	2 ms 6,548 KB
testcase_13	AC	2 ms 6,548 KB
testcase_14	AC	2 ms 6,548 KB
testcase_15	AC	2 ms 6,548 KB
testcase_16	AC	1 ms 6,548 KB
testcase_17	AC	1 ms 6,548 KB
testcase_18	AC	2 ms 6,548 KB
testcase_19	AC	3 ms 6,548 KB
testcase_20	AC	4 ms 6,548 KB
testcase_21	AC	4 ms 6,548 KB
testcase_22	AC	3 ms 6,548 KB
testcase_23	AC	3 ms 6,548 KB
testcase_24	AC	4 ms 6,548 KB
testcase_25	AC	3 ms 6,548 KB
testcase_26	AC	3 ms 6,548 KB
testcase_27	AC	3 ms 6,548 KB
testcase_28	AC	4 ms 6,548 KB
testcase_29	AC	2 ms 6,548 KB
testcase_30	AC	2 ms 6,548 KB
testcase_31	AC	2 ms 6,548 KB
testcase_32	AC	2 ms 6,548 KB
testcase_33	AC	2 ms 6,548 KB
testcase_34	AC	2 ms 6,548 KB
testcase_35	AC	2 ms 6,548 KB
testcase_36	AC	2 ms 6,548 KB
testcase_37	AC	2 ms 6,548 KB
testcase_38	AC	2 ms 6,548 KB
testcase_39	AC	28 ms 7,808 KB
testcase_40	AC	28 ms 7,808 KB
testcase_41	AC	28 ms 7,936 KB
testcase_42	AC	29 ms 7,936 KB
testcase_43	AC	28 ms 7,808 KB
testcase_44	AC	28 ms 7,808 KB
testcase_45	AC	28 ms 7,808 KB
testcase_46	AC	28 ms 7,808 KB
testcase_47	AC	28 ms 7,808 KB
testcase_48	AC	28 ms 7,936 KB
testcase_49	AC	76 ms 27,440 KB
testcase_50	AC	93 ms 28,908 KB
testcase_51	AC	98 ms 31,116 KB
testcase_52	AC	72 ms 25,672 KB
testcase_53	AC	98 ms 31,820 KB
testcase_54	AC	82 ms 28,004 KB
testcase_55	AC	136 ms 45,784 KB
testcase_56	AC	79 ms 26,920 KB
testcase_57	AC	75 ms 28,036 KB
testcase_58	AC	103 ms 32,556 KB
testcase_59	AC	30 ms 8,064 KB
testcase_60	AC	30 ms 8,064 KB
testcase_61	AC	29 ms 7,808 KB
testcase_62	AC	29 ms 7,936 KB
testcase_63	AC	30 ms 8,192 KB
testcase_64	AC	29 ms 7,936 KB
testcase_65	AC	29 ms 8,064 KB
testcase_66	AC	29 ms 7,936 KB
testcase_67	AC	29 ms 7,936 KB
testcase_68	AC	28 ms 7,936 KB
testcase_69	AC	288 ms 85,740 KB
testcase_70	AC	2 ms 6,548 KB
testcase_71	AC	2 ms 6,548 KB
testcase_72	AC	1 ms 6,548 KB

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

ソースコード

raw source code

import macros;macro ImportExpand(s:untyped):untyped = parseStmt($s[2])
# {.checks: off.}
ImportExpand "cplib/tmpl/citrus.nim" <=== "when not declared CPLIB_TMPL_CITRUS:\n    const CPLIB_TMPL_CITRUS* = 1\n    {.warning[UnusedImport]: off.}\n    {.hint[XDeclaredButNotUsed]: off.}\n    import os\n    import algorithm\n    import sequtils\n    import tables\n    import macros\n    import std/math\n    import sets\n    import strutils\n    import strformat\n    import sugar\n    import streams\n    import deques\n    import bitops\n    import heapqueue\n    const MODINT998244353* = 998244353\n    const MODINT1000000007* = 1000000007\n    const INF* = 100100111\n    const INFL* = int(3300300300300300491)\n    type double* = float64\n    let readNext = iterator(getsChar: bool = false): string {.closure.} =\n        while true:\n            var si: string\n            try: si = stdin.readLine\n            except EOFError: yield \"\"\n            for s in si.split:\n                if getsChar:\n                    for i in 0..<s.len(): yield s[i..i]\n                else:\n                    yield s\n    proc input*(t: typedesc[string]): string = readNext()\n    proc input*(t: typedesc[char]): char = readNext(true)[0]\n    proc input*(t: typedesc[int]): int = readNext().parseInt\n    proc input*(t: typedesc[float]): float = readNext().parseFloat\n    macro input*(t: typedesc, n: varargs[int]): untyped =\n        var repStr = \"\"\n        for arg in n:\n            repStr &= &\"({arg.repr}).newSeqWith \"\n        parseExpr(&\"{repStr}input({t})\")\n    macro input*(ts: varargs[auto]): untyped =\n        var tupStr = \"\"\n        for t in ts:\n            tupStr &= &\"input({t.repr}),\"\n        parseExpr(&\"({tupStr})\")\n    macro input*(n: int, ts: varargs[auto]): untyped =\n        for typ in ts:\n            if typ.typeKind != ntyAnything:\n                error(\"Expected typedesc, got \" & typ.repr, typ)\n        parseExpr(&\"({n.repr}).newSeqWith input({ts.repr})\")\n    proc `fmtprint`*(x: int or string or char): string = return $x\n    proc `fmtprint`*(x: float or float32 or\n            float64): string = return &\"{x:.16f}\"\n    proc `fmtprint`*[T](x: seq[T] or Deque[T] or HashSet[T] or set[\n            T]): string = return x.toSeq.join(\" \")\n    proc `fmtprint`*[T, N](x: array[T, N]): string = return x.toSeq.join(\" \")\n    proc `fmtprint`*[T](x: HeapQueue[T]): string =\n        var q = x\n        while q.len != 0:\n            result &= &\"{q.pop()}\"\n            if q.len != 0: result &= \" \"\n    proc `fmtprint`*[T](x: CountTable[T]): string =\n        result = x.pairs.toSeq.mapIt(&\"{it[0]}: {it[1]}\").join(\" \")\n    proc `fmtprint`*[K, V](x: Table[K, V]): string =\n        result = x.pairs.toSeq.mapIt(&\"{it[0]}: {it[1]}\").join(\" \")\n    proc print*(prop: tuple[f: File, sepc: string, endc: string, flush: bool],\n            args: varargs[string, `fmtprint`]) =\n        for i in 0..<len(args):\n            prop.f.write(&\"{args[i]}\")\n            if i != len(args) - 1: prop.f.write(prop.sepc) else: prop.f.write(prop.endc)\n        if prop.flush: prop.f.flushFile()\n    proc print*(args: varargs[string, `fmtprint`]) = print((f: stdout,\n            sepc: \" \", endc: \"\\n\", flush: false), args)\n    proc inner_debug*(x: auto) = print((f: stderr, sepc: \"\", endc: \"\",\n            flush: true), x)\n    const LOCAL_DEBUG{.booldefine.} = false\n    macro debug*(n: varargs[typed]): untyped =\n        when LOCAL_DEBUG:\n            result = newNimNode(nnkStmtList, n)\n            for i in 0..n.len-1:\n                if n[i].kind == nnkStrLit:\n                    result.add(newCall(\"inner_debug\", n[i]))\n                    result.add(newCall(\"inner_debug\", newStrLitNode(\": \")))\n                    result.add(newCall(\"inner_debug\", n[i]))\n                else:\n                    result.add(newCall(\"inner_debug\", toStrLit(n[i])))\n                    result.add(newCall(\"inner_debug\", newStrLitNode(\": \")))\n                    result.add(newCall(\"inner_debug\", n[i]))\n                if i != n.len-1:\n                    result.add(newCall(\"inner_debug\", newStrLitNode(\", \")))\n                else:\n                    result.add(newCall(\"inner_debug\", newStrLitNode(\"\\n\")))\n        else:\n            return quote do:\n                discard\n    proc `%`*(x: SomeInteger, y: SomeInteger): int = (((x mod y) + y) mod y)\n    proc `//`*(x: int, y: int): int = ((x - (x%y)) div y)\n    proc `^`*(x: int, y: int): int = x xor y\n    proc `&`*(x: int, y: int): int = x and y\n    proc `|`*(x: int, y: int): int = x or y\n    proc `>>`*(x: int, y: int): int = x shr y\n    proc `<<`*(x: int, y: int): int = x shl y\n    proc `%=`*(x: var SomeInteger or int64, y: SomeInteger or\n            int64): void = x = x % y\n    proc `//=`*(x: var int, y: int): void = x = x // y\n    proc `^=`*(x: var int, y: int): void = x = x ^ y\n    proc `&=`*(x: var int, y: int): void = x = x & y\n    proc `|=`*(x: var int, y: int): void = x = x | y\n    proc `>>=`*(x: var int, y: int): void = x = x >> y\n    proc `<<=`*(x: var int, y: int): void = x = x << y\n    proc `[]`*(x: int, n: int): bool = (x and (1 shl n)) != 0\n\n    proc pow*(a, n: int, m = INFL): int =\n        var\n            rev = 1\n            a = a\n            n = n\n        while n > 0:\n            if n % 2 != 0: rev = (rev * a) mod m\n            if n > 1: a = (a * a) mod m\n            n >>= 1\n        return rev\n    proc sqrt*(x: int): int =\n        assert(x >= 0)\n        result = int(sqrt(float64(x)))\n        while result * result > x: result -= 1\n        while (result+1) * (result+1) <= x: result += 1\n    proc chmax*[T](x: var T, y: T): bool = (if x < y: (x = y; return true;\n        ) return false)\n    proc chmin*[T](x: var T, y: T): bool = (if x > y: (x = y; return true;\n        ) return false)\n    proc `max=`*[T](x: var T, y: T) = x = max(x, y)\n    proc `min=`*[T](x: var T, y: T) = x = min(x, y)\n    proc at*(x: char, a = '0'): int = int(x) - int(a)\n    converter tofloat*(n: int): float = float(n)\n    iterator rangeiter*(start: int, ends: int, step: int): int =\n        var i = start\n        if step < 0:\n            while i > ends:\n                yield i\n                i += step\n        elif step > 0:\n            while i < ends:\n                yield i\n                i += step\n    iterator rangeiter*(ends: int): int = (for i in 0..<ends: yield i)\n    iterator rangeiter*(start: int, ends: int): int = (for i in\n            start..<ends: yield i)\n    proc Yes*(b: bool = true): void = print(if b: \"Yes\" else: \"No\")\n    proc No*(b: bool = true): void = Yes(not b)\n    proc YES_upper*(b: bool = true): void = print(if b: \"YES\" else: \"NO\")\n    proc NO_upper*(b: bool = true): void = Yes_upper(not b)\n    const DXY* = [(0, -1), (0, 1), (-1, 0), (1, 0)]\n    const DDXY* = [(1, -1), (1, 0), (1, 1), (0, -1), (0, 1), (-1, -1), (-1, 0),\n            (-1, 1)]\n    macro exit*(statement: untyped): untyped =\n        quote do:\n            `statement`\n            quit()\n    proc vector*[T](d1, : int, default: T = T(0)): seq[T] = newSeqWith(d1, default)\n    proc vv*[T](d1, d2: int, default: T = T(0)): seq[seq[T]] = newSeqWith(d1,\n            newSeqWith(d2, default))\n    proc vvv*[T](d1, d2, d3: int, default: T = T(0)): seq[seq[seq[\n            T]]] = newSeqWith(d1, newSeqWith(d2, newSeqWith(d3, default)))\n    proc vvvv*[T](d1, d2, d3, d4: int, default: T = T(0)): seq[seq[seq[seq[\n            T]]]] = newSeqWith(d1, newSeqWith(d2, newSeqWith(d3, newSeqWith(d4, default))))\n    proc vvvvv*[T](d1, d2, d3, d4, d5: int, default: T = T(0)): seq[seq[seq[seq[\n            seq[T]]]]] = newSeqWith(d1, newSeqWith(d2, newSeqWith(d3,\n            newSeqWith(d4, newSeqWith(d5, default)))))\n    proc vvvvvv*[T](d1, d2, d3, d4, d5, d6: int, default: T = T(0)): seq[seq[\n            seq[seq[seq[seq[T]]]]]] = newSeqWith(d1, newSeqWith(d2, newSeqWith(\n            d3, newSeqWith(d4, newSeqWith(d5, newSeqWith(d6, default))))))\n    discard\n"


type Node = object
    link: seq[ref Node]
    suffix_link: ref Node
    len, count, id: int

type PalindromicTree = object
    amax: int
    nodes: seq[ref Node]

proc newNode(pt: var PalindromicTree, amax, len: int): ref Node =
    result = new Node
    result[].link = newSeq[ref Node](amax)
    result[].suffix_link = nil
    result[].len = len
    result[].count = 0
    result[].id = pt.nodes.len
    pt.nodes.add(result)

proc init(amax: int): PalindromicTree =
    discard result.newNode(amax, -1)
    discard result.newNode(amax, 0)
    result.amax = amax
    # result.nodes[0][].suffix_link = result.nodes[0]
    result.nodes[1][].suffix_link = result.nodes[0]


proc initPalindromicTree(a: seq[int], amax: int = -1): PalindromicTree =
    var amax = amax
    if amax < 0: amax = a.max
    result = init(amax)
    proc find_longest(pos: int, node: ref Node): ref Node =
        var ln = pos - node[].len - 1
        if ln >= 0 and a[ln] == a[pos]:
            return node
        return find_longest(pos, node[].suffix_link)
    var current_node = result.nodes[0]
    for i in 0..<a.len:
        current_node = find_longest(i, current_node)
        # echo current_node[], " ", a[i]
        if current_node[].link[a[i]] == nil:
            current_node[].link[a[i]] = result.newNode(amax, current_node[].len + 2)
        if current_node == result.nodes[0]:
            current_node[].link[a[i]][].suffix_link = result.nodes[1]
        else:
            current_node[].link[a[i]][].suffix_link = find_longest(i, current_node[].suffix_link)[].link[a[i]]
        current_node = current_node[].link[a[i]]
        current_node[].count += 1

proc initPalindromicTree(s: string, c: char = 'a'): PalindromicTree =
    return initPalindromicTree(s.mapIt(int(it) - int(c)), 26)

proc get_palindrome(pt: PalindromicTree, node: ref Node): seq[int] =
    if node == pt.nodes[0] or node == pt.nodes[1]: return newSeq[int](0)
    var ans = newSeq[int](0)
    proc dfs(x: ref Node): bool =
        if x == node: return true
        for i in 0..<pt.amax:
            if x[].link[i] == nil: continue
            if dfs(x[].link[i]):
                ans.add(i)
                return true
        return false
    discard dfs(pt.nodes[0])
    discard dfs(pt.nodes[1])
    for i in ans.len..<node[].len:
        ans.add(ans[node[].len-1-i])
    return ans

proc update_count(pt: PalindromicTree) =
    for node in pt.nodes[1..<pt.nodes.len].reversed:
        node[].suffix_link[].count += node[].count

var s = input(string).reversed.join("")
var pt = initPalindromicTree(s)
pt.update_count

# proc dfs(x: ref Node) =
#     print(pt.get_palindrome(x), x[].count)
#     for i in 0..<pt.amax:
#         if x[].link[i] == nil: continue
#         dfs(x[].link[i])
#     echo "end"
# dfs(pt.nodes[0])
# dfs(pt.nodes[1])

# for node in pt.nodes:
#     print(pt.get_palindrome(node), ":", node[].count)
# updete_count(pt)

var ans = 0
var score = newSeqWith(pt.nodes.len, 0)
for i in 1..<pt.nodes.len:
    var node = pt.nodes[i]
    score[i] = score[node[].suffix_link[].id] + node[].count * node[].len
print(score.max)

yukicoder

結果

テストケース

ソースコード