結果

問題 No.261 ぐるぐるぐるぐる！あみだくじ！
ユーザー shotoyooshotoyoo
提出日時 2021-06-14 22:47:00
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 83 ms / 5,000 ms
コード長 3,267 bytes
コンパイル時間 230 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 68,864 KB
最終ジャッジ日時 2024-12-25 03:11:08
合計ジャッジ時間 4,387 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda : sys.stdin.readline().rstrip()

sys.setrecursionlimit(2*10**5+10)
write = lambda x: sys.stdout.write(x+"\n")
debug = lambda x: sys.stderr.write(x+"\n")
writef = lambda x: print("{:.12f}".format(x))

def part(p):
    n = len(p)
    done = [0]*n
    ans = []
    for i in range(n):
        if done[i]:
            continue
        l = [i]
        i0 = i
        done[i] = 1
        while p[i]!=i0:
            i = p[i]
            l.append(i)
            done[i] = 1
        ans.append(l)
    return ans
def gcd2(a, b):
    """a*x + b*y = gcd(a,b)なるx,yも求める
    """
    l = []
    while b:
        l.append(divmod(a,b))
        a, b = b, a%b
    x, y = 1, 0
    for aa,bb in l[::-1]:
        x, y = y, x - aa*y
    return a, x, y

def modinv(x, M):
    """素数ではないM、Mと互いに素なxに対し
    x * y == 1 mod M なるyを求める
    """
    a,xx,yy = gcd2(x,M)
    return a,xx%M

def solve(a,b,n):
    """a*x = b mod nを解く
    """
    g,xx,yy = gcd2(a,n)
    if b%g!=0:
        return None
    a //= g
    n //= g
    b //= g
    # aとnが互いに素になっているので
    ainv = modinv(a, n)[1]
    x = ainv*b%n
    return x

def crt(rs, ms):
    """中国剰余定理
    x == rs[i] mod ms[i] をみたすxをを求め、(x, l(=ms))を返す
    (そのようなxは x + i * lとして書ける)
    存在しない場合はNoneを返す
    長さが0の配列を渡すと(0,1)を返す
    """
    r0 = 0
    m0 = 1
    for r1,m1 in zip(rs, ms):
        if m0<m1:
            m0, m1 = m1, m0
            r0, r1 = r1, r0
        if m0%m1==0:
            if r0%m1 != r1:
                return None,None
            else:
                continue
#         print(m0,m1)
        g,im = modinv(m0, m1)
        u1 = m1//g
        if (r1-r0)%g!=0:
            return None,None
        x = (r1-r0) // g % u1 * im % u1
        r0 += x * m0
        m0 *= u1
        if r0<0:
            r0 += m0
    return r0,m0

n = int(input())
k = int(input())
p = list(range(n))
ps = [list(range(n))]
for i in range(k):
    x,y = list(map(int, input().split()))
    x -= 1
    y -= 1
    p[x],p[y] = p[y], p[x]
ps.append(p[:])
for i in range(100):
    ps.append([p[i] for i in ps[-1]])
res = part(p)
s = [set(item) for item in res]
m = len(res)
g = [-1]*n
for i,item in enumerate(res):
    for v in item:
        g[v] = i
q = int(input())
ans = []
from collections import deque
for i in range(q):
    a = list(map(lambda i: int(i)-1, input().split()))
    l = [[] for _ in range(m)]
    for j in range(n):
        l[g[j]].append(a[j])
    ms = []
    rs = []
    for j in range(m):
        v = l[j][0]
        if v not in res[j]:
            ans.append(-1)
            break
        index = [jj for jj in range(n) if g[jj]==j]
        for ind in range(100):
            tmp = [ps[ind][jj] for jj in index]
            if tmp==l[j]:
                break
        else:
            ans.append(-1)
            break
        rs.append(ind)
        ms.append(len(res[j]))
    else:
        r0, m0 = crt(rs, ms)
        if r0 is None:
            ans.append("-1")
        else:
            val = r0%m0
            if val==0:
                val = m0
            ans.append(val)
#     break
write("\n".join(map(str, ans)))
0