結果

問題 No.2858 Make a Palindrome
ユーザー kemunikukemuniku
提出日時 2024-08-31 02:19:58
言語 Nim
(2.0.2)
結果
WA  
実行時間 -
コード長 12,544 bytes
コンパイル時間 4,225 ms
コンパイル使用メモリ 92,016 KB
実行使用メモリ 14,984 KB
最終ジャッジ日時 2024-08-31 02:20:12
合計ジャッジ時間 13,815 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 481 ms
10,756 KB
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 AC 150 ms
14,176 KB
testcase_24 WA -
testcase_25 WA -
testcase_26 AC 114 ms
13,400 KB
testcase_27 WA -
testcase_28 WA -
testcase_29 AC 159 ms
14,352 KB
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
権限があれば一括ダウンロードができます

ソースコード

diff #

import macros;macro ImportExpand(s:untyped):untyped = parseStmt($s[2])
import macros
import macros
ImportExpand "cplib/tmpl/sheep.nim" <=== "when not declared CPLIB_TMPL_SHEEP:\n    const CPLIB_TMPL_SHEEP* = 1\n    {.warning[UnusedImport]: off.}\n    {.hint[XDeclaredButNotUsed]: off.}\n    import algorithm\n    import sequtils\n    import tables\n    import macros\n    import math\n    import sets\n    import strutils\n    import strformat\n    import sugar\n    import heapqueue\n    import streams\n    import deques\n    import bitops\n    import std/lenientops\n    import options\n    #入力系\n    proc scanf(formatstr: cstring){.header: \"<stdio.h>\", varargs.}\n    proc getchar(): char {.importc: \"getchar_unlocked\", header: \"<stdio.h>\", discardable.}\n    proc ii(): int {.inline.} = scanf(\"%lld\\n\", addr result)\n    proc lii(N: int): seq[int] {.inline.} = newSeqWith(N, ii())\n    proc si(): string {.inline.} =\n        result = \"\"\n        var c: char\n        while true:\n            c = getchar()\n            if c == ' ' or c == '\\n':\n                break\n            result &= c\n    #chmin,chmax\n    template `max=`(x, y) = x = max(x, y)\n    template `min=`(x, y) = x = min(x, y)\n    #bit演算\n    proc `%`*(x: int, y: int): int =\n        result = x mod y\n        if y > 0 and result < 0: result += y\n        if y < 0 and result > 0: result += y\n    proc `//`*(x: int, y: int): int{.inline.} =\n        result = x div y\n        if y > 0 and result * y > x: result -= 1\n        if y < 0 and result * y < x: result -= 1\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, y: int): int = x^y\n    proc `**=`(x: var int, y: int): void = x = x^y\n    proc `^`(x: int, y: int): int = x xor y\n    proc `|`(x: int, y: int): int = x or y\n    proc `&`(x: int, y: int): int = x and y\n    proc `>>`(x: int, y: int): int = x shr y\n    proc `<<`(x: int, y: int): int = x shl y\n    proc `~`(x: int): int = not x\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 `!`(x: char, a = '0'): int = int(x)-int(a)\n    #定数\n    #[ include cplib/utils/constants ]#\n    when not declared CPLIB_UTILS_CONSTANTS:\n        const CPLIB_UTILS_CONSTANTS* = 1\n        const INF32*: int32 = 100100111.int32\n        const INF64*: int = int(3300300300300300491)\n    const INF = INF64\n    #converter\n\n    #range\n    iterator range(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 range(ends: int): int = (for i in 0..<ends: yield i)\n    iterator range(start: int, ends: int): int = (for i in\n            start..<ends: yield i)\n\n    #joinが非stringでめちゃくちゃ遅いやつのパッチ\n    proc join*[T: not string](a: openArray[T], sep: string = \"\"): string = a.mapit($it).join(sep)\n"
ImportExpand "cplib/str/can_reverse_hash_string.nim" <=== "when not declared CPLIB_STR_HASHSTRING:\n    const CPLIB_STR_HASHSTRING* = 1\n    # hash_stringといっしょに使われるのは想定していないのであえてconstを衝突させています\n    import random\n    type HashString* =object\n        hash* :uint\n        rhash* :uint\n        bpow  :uint\n        size: int\n    const MASK30 = (1u shl 30) - 1\n    const MASK31 = (1u shl 31) - 1\n    const RH_MOD = (1u shl 61) - 1\n    const POW_CALC = 500000\n\n    randomize()\n\n    proc calc_mod(x: uint): uint =\n        result = (x shr 61) + (x and RH_MOD)\n        if result >= RH_MOD:\n            result -= RH_MOD\n\n    proc mul(a, b: uint): uint =\n        let\n            a_upper = a shr 31\n            a_lower = a and MASK31\n            b_upper = b shr 31\n            b_lower = b and MASK31\n            mid = a_lower * b_upper + a_upper * b_lower\n            mid_upper = mid shr 30\n            mid_lower = mid and MASK30\n        result = a_upper * b_upper * 2 + mid_upper + (mid_lower shl 31) + a_lower * b_lower\n\n\n    proc inner_pow(a:uint, n: int): uint =\n        var a = a\n        var n = n\n        result = 1\n        while n > 0:\n            if (n and 1) != 0:\n                result = mul(result, a).calc_mod\n            a = mul(a, a).calc_mod\n            n = n shr 1\n\n    let hashstring_base:uint = rand(129u..(1u shl 30))\n    let inv_hashstring_base:uint = inner_pow(hashstring_base,int(RH_MOD)-2)\n    var pows : seq[uint] = newseq[uint](POW_CALC+1)\n    var invpows : seq[uint] = newseq[uint](POW_CALC+1)\n    pows[0] = 1\n    invpows[0] = 1\n    for i in 1..POW_CALC:\n        pows[i] = (mul(pows[i-1],hashstring_base).calc_mod)\n        invpows[i] = (mul(invpows[i-1],inv_hashstring_base).calc_mod)\n\n    proc base_pow(n:int):uint=\n        if n >= len(pows):\n            return inner_pow(hashstring_base,n)\n        else:\n            return pows[n]\n    \n    proc base_inv_pow(n:int):uint=\n        if n >= len(invpows):\n            return inner_pow(inv_hashstring_base,n)\n        else:\n            return invpows[n]\n\n    proc tohash*(S:string):HashString=\n        var hash = 0u\n        var rhash = 0u\n        var tmp = 1u\n        for i in countdown(len(S)-1,0,1):\n            hash = (hash+mul(uint(int(S[i])),tmp)).calc_mod\n            rhash = (rhash+mul(uint(int(S[len(S)-1-i])),tmp)).calc_mod\n            tmp = mul(tmp,hashstring_base).calc_mod \n        result = HashString(hash:hash,rhash:rhash,bpow:base_pow(len(S)),size:len(S))\n\n    proc tohash*(S:char):HashString=\n        result = HashString(hash:uint(int(S)),rhash:uint(int(S)),bpow:hashstring_base,size:1)\n\n    proc `&`*(L,R:HashString):HashString=\n        result = HashString(hash:(mul(L.hash,R.bpow).calc_mod+R.hash).calc_mod,\n                            rhash:(mul(R.rhash,L.bpow).calc_mod+L.rhash).calc_mod,\n                            bpow:mul(L.bpow,R.bpow).calc_mod,size:L.size+R.size)\n\n    proc `==`*(L,R:HashString):bool=\n        return (L.size == R.size) and (L.hash == R.hash)\n\n    proc len*(H:HashString):int=int(H.size)\n\n    proc `*`*(H:HashString,x:int):HashString=\n        var\n            size = H.size * x\n            bpow = uint(1)\n            tmp_hash = H.hash\n            tmp_rhash = H.rhash\n            tmp_b = H.bpow\n            hash = uint(0)\n            rhash = uint(0)\n            x = x\n        while x > 0:\n            if x mod 2 != 0:\n                hash = (mul(hash,tmp_b).calc_mod+tmp_hash).calc_mod\n                rhash = (mul(tmp_rhash,bpow).calc_mod+rhash).calc_mod\n                bpow = mul(bpow,tmp_b).calc_mod\n            if x > 1:\n                tmp_hash = (mul(tmp_hash,tmp_b).calc_mod+tmp_hash).calc_mod\n                tmp_rhash = (mul(tmp_rhash,tmp_b).calc_mod+tmp_rhash).calc_mod\n                tmp_b = mul(tmp_b,tmp_b).calc_mod\n            x = x shr 1\n        return HashString(hash:hash,rhash:rhash,bpow:bpow,size:size)\n    \n    proc isPalindrome*(H:HashString):bool=\n        H.hash == H.rhash\n\n    # proc removePrefix*(H,prefix:HashString):HashString=\n    #     var hash = (H.hash + (RH_MOD - mul(prefix.hash,base_pow(len(H)-len(prefix))).calc_mod)).calc_mod\n    #     var l = len(H)-len(prefix)\n    #     return HashString(hash:hash,bpow:base_pow(l),size:l)\n\n    type RollingHashBase = ref object\n        S : string\n        prefixs : seq[uint]\n        suffixs : seq[uint]\n        size : int \n\n    type RollingHash* = object\n        R : RollingHashBase\n        l : int\n        r : int\n\n    proc len*(S:RollingHashBase):int=\n        return int(S.size)\n\n    proc len*(S:RollingHash):int=\n        return int(S.r-S.l)\n\n    proc get_substring(R:RollingHashBase,l,r:int):RollingHash=\n        # 半開区間とする。\n        # 空文字列用にr=0も許容していることに注意。\n        # 空文字列はl=0,r=0のみ許容している。\n        assert l in 0..<R.size and r in 0..R.size and (l < r or (l == 0 and r == 0))\n        result.R = R\n        result.l = l\n        result.r = r\n\n    proc `[]`*(R:RollingHashBase,slice:HSlice[int,int]):RollingHash=\n        assert slice.a >= 0 and slice.b >= 0\n        return R.get_substring(slice.a,slice.b+1)\n\n\n    proc `[]`*(S:RollingHash,slice:HSlice[int,int]):RollingHash=\n        if len(slice) == 0:\n            return S.R.get_substring(0,0)\n        assert slice.a in 0..<len(S) and slice.b in 0..<len(S)\n        return S.R.get_substring(S.l+slice.a,S.l+slice.b+1)\n\n    proc gethash(S:RollingHash,slice:HSlice[int,int]):uint=\n        return (S.R.prefixs[(S.l+slice.b+1)] + (RH_MOD - mul(S.R.prefixs[S.l+slice.a],base_pow(((S.l+slice.b+1)-(S.l+slice.a)))).calc_mod)).calc_mod\n\n\n    proc `[]`*(S:RollingHash,idx:int):char=\n        return S.R.S[idx+int(S.l)]\n\n    proc initRollingHash*(S:string):RollingHash=\n        var rolling = RollingHashBase()\n        rolling.S = S\n        rolling.prefixs = newSeq[uint](len(S)+1)\n        rolling.suffixs = newSeq[uint](len(S)+1)\n        rolling.suffixs[^1] = 0\n        rolling.prefixs[0] = 0\n        var tmp = 1u\n        for i in 1..len(S):\n            rolling.prefixs[i] = (mul(rolling.prefixs[i-1],hashstring_base) + uint(int(S[i-1]))).calc_mod()\n            rolling.suffixs[len(S)-i] = (mul(rolling.suffixs[len(S)-i+1],hashstring_base) + uint(int(S[len(S)-i]))).calc_mod()\n            tmp = mul(tmp,hashstring_base).calc_mod\n        rolling.size = (len(S))\n        return rolling[0..<len(S)]\n\n\n\n    converter toHashString*(self:RollingHash):HashString=\n        return HashString(  hash:(self.R.prefixs[self.r] + (RH_MOD - mul(self.R.prefixs[self.l],base_pow(self.r-self.l)).calc_mod)).calc_mod,\n                            rhash:mul((self.R.suffixs[self.l] + (RH_MOD - self.R.suffixs[self.r])).calc_mod,base_inv_pow(len(self.R)-self.r)).calc_mod,\n                            bpow:base_pow(self.r-self.l),\n                            size:self.r-self.l)\n\n    proc`$`*(S:RollingHash):string=\n        return S.R.S[S.l..<S.r]\n\n    proc `==`*(S,T:RollingHash):bool=\n        return len(S) == len(T) and (S.R.prefixs[S.r] + (RH_MOD - mul(S.R.prefixs[S.l],base_pow(S.r-S.l)).calc_mod)).calc_mod == \n            (T.R.prefixs[T.r] + (RH_MOD - mul(T.R.prefixs[T.l],base_pow(T.r-T.l)).calc_mod)).calc_mod\n\n    proc LCP*(S,T:RollingHash):int=\n        var ok = 0\n        var ng = min(len(S),len(T))+1\n        while ng-ok > 1:\n            var mid = (ok + ng) div 2\n            if S.gethash(0..<mid) == T.gethash(0..<mid): ok = mid\n            else: ng = mid\n        return ok\n\n    proc cmp*(S,T:RollingHash):int=\n        var S = S\n        var T = T\n        var flg = 1\n        if len(S) > len(T):\n            swap(S,T)\n            flg *= -1\n        var lcp = LCP(S,T)\n        if len(S) == lcp:\n            if len(S) == len(T):\n                return 0\n            else:\n                return -1*flg\n        else:\n            if S[lcp] < T[lcp]:\n                return -1*flg\n            else:\n                return flg\n\n    proc `<`*(S,T:RollingHash):bool=\n        return cmp(S,T) < 0\n\n    \n"
var T = ii()

for _ in range(T):
    var N,M = ii()
    var S = si()
    #Sを無限個連結した文字列について、
    #1.あるiからM文字連続のものが回文かを判定する。
    #2.あるiからM+1文字連続のものが回文かをはんていする。
    var ans = INF
    var HS = S.initRollingHash()

    proc solve(i,x:int):bool=
        if (i+x) < N:
            #echo "!",HS[i..<(i+x)].isPalindrome(),HS[i..<(i+x)],HS[i..<(i+x)].tohashString()
            return HS[i..<(i+x)].isPalindrome()
        var tmp1 = HS[i..<N].toHashString()
        tmp1 = tmp1 & (HS * ((x-len(i..<N)) // N))
        var nokori = x-(((x-len(i..<N)) // N)*N + len(i..<N))
        tmp1 = tmp1 & HS[0..<nokori]
        #echo tmp1
        return tmp1.isPalindrome()


    for i in range(N):
        if solve(i,M):
            ans.min = (M-(N-i)+(N-1))//N + 1
        if solve(i,M+1):
            ans.min = (M+1-(N-i)+(N-1))//N + 1

    if ans == INF:
        stdout.writeLine -1
    else:
        stdout.writeLine ans
0