#834482 (PyPy3) No.2197 Same Dish

提出ソース

結果

問題	No.2197 Same Dish
ユーザー	Keroru
提出日時	2023-01-20 22:23:50
言語	PyPy3 (7.3.15)
結果	WA
実行時間	-
コード長	5,634 bytes
コンパイル時間	458 ms
コンパイル使用メモリ	82,272 KB
実行使用メモリ	849,584 KB
最終ジャッジ日時	2024-06-23 10:21:01
合計ジャッジ時間	4,818 ms
ジャッジサーバーID （参考情報）	judge3 / judge4

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	160 ms 94,176 KB
testcase_01	AC	157 ms 94,136 KB
testcase_02	AC	156 ms 94,000 KB
testcase_03	WA	-
testcase_04	AC	153 ms 94,092 KB
testcase_05	AC	154 ms 93,920 KB
testcase_06	AC	153 ms 94,076 KB
testcase_07	AC	161 ms 94,108 KB
testcase_08	AC	153 ms 94,208 KB
testcase_09	AC	149 ms 93,972 KB
testcase_10	AC	154 ms 94,028 KB
testcase_11	AC	154 ms 94,052 KB
testcase_12	WA	-
testcase_13	MLE	-
testcase_14	--	-
testcase_15	--	-
testcase_16	--	-
testcase_17	--	-
testcase_18	--	-
testcase_19	--	-
testcase_20	--	-
testcase_21	--	-
testcase_22	--	-

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

ソースコード

raw source code

import sys
read=sys.stdin.buffer.read;readline=sys.stdin.buffer.readline;input=lambda:sys.stdin.readline().rstrip()
import bisect,string,math,time,functools,random,fractions
from bisect import*
from heapq import heappush,heappop,heapify
from collections import deque,defaultdict,Counter
from itertools import permutations,combinations,groupby
import itertools
rep=range;R=range
def I():return int(input())
def LI():return [int(i) for i in input().split()]
def SLI():return sorted([int(i) for i in input().split()])
def LI_():return [int(i)-1 for i in input().split()]
def S_():return input()
def IS():return input().split()
def LS():return [i for i in input().split()]
def NI(n):return [int(input()) for i in range(n)]
def NI_(n):return [int(input())-1 for i in range(n)]
def NLI(n):return [[int(i) for i in input().split()] for i in range(n)]
def NLI_(n):return [[int(i)-1 for i in input().split()] for i in range(n)]
def StoLI():return [ord(i)-97 for i in input()] 
def ItoS(n):return chr(n+97)
def LtoS(ls):return ''.join([chr(i+97) for i in ls])
def RLI(n=8,a=1,b=10):return [random.randint(a,b)for i in range(n)]
def RI(a=1,b=10):return random.randint(a,b)
def GI(V,E,ls=None,Directed=False,index=1):
    org_inp=[];g=[[] for i in range(V)]
    FromStdin=True if ls==None else False
    for i in range(E):
        if FromStdin:
            inp=LI()
            org_inp.append(inp)
        else:
            inp=ls[i]
        if len(inp)==2:a,b=inp;c=1
        else:a,b,c=inp
        if index==1:a-=1;b-=1
        aa=a,c,;bb=b,c,;g[a].append(bb)
        if not Directed:g[b].append(aa)
    return g,org_inp
def RE(E):
    rt=[[]for i in range(len(E))]
    for i in range(len(E)):
        for nb,d in E[i]:
            rt[nb]+=(i,d),
    return rt
def RLE(it):
    rt=[]
    for i in it:
        if rt and rt[-1][0]==i:rt[-1][1]+=1
        else:rt+=[i,1],
    return rt
def GGI(h,w,search=None,replacement_of_found='.',mp_def={'#':1,'.':0},boundary=1):
    #h,w,g,sg=GGI(h,w,search=['S','G'],replacement_of_found='.',mp_def={'#':1,'.':0},boundary=1) # sample usage
    mp=[boundary]*(w+2);found={}
    for i in R(h):
        s=input()
        for char in search:
            if char in s:
                found[char]=((i+1)*(w+2)+s.index(char)+1)
                mp_def[char]=mp_def[replacement_of_found]
        mp+=[boundary]+[mp_def[j] for j in s]+[boundary]
    mp+=[boundary]*(w+2)
    return h+2,w+2,mp,found
def TI(n):return GI(n,n-1)
def accum(ls):
    rt=[0]
    for i in ls:rt+=[rt[-1]+i]
    return rt
def bit_combination(n,base=2):
    rt=[]
    for tb in R(base**n):s=[tb//(base**bt)%base for bt in R(n)];rt+=[s]
    return rt
def gcd(x,y):
    if y==0:return x
    if x%y==0:return y
    while x%y!=0:x,y=y,x%y
    return y
def YN(x):print(['NO','YES'][x])
def Yn(x):print(['No','Yes'][x])
def show(*inp,end='\n'):
    if show_flg:print(*inp,end=end)

inf=float('inf')
FourNb=[(-1,0),(1,0),(0,1),(0,-1)];EightNb=[(-1,0),(1,0),(0,1),(0,-1),(1,1),(-1,-1),(1,-1),(-1,1)];compas=dict(zip('WENS',FourNb));cursol=dict(zip('UDRL',FourNb));HexNb=[(-1,0),(-1,-1),(0,1),(0,-1),(1,1),(1,0)]
alp=[chr(ord('a')+i)for i in range(26)]
#sys.setrecursionlimit(10**7)

def gcj(t,*a):
    print('Case #{}:'.format(t+1),*a)

def INP():
    N=80
    n=random.randint(1,N)
    x=random.randint(1,N)
    return n,x
def Rtest(T):
    case,err=0,0
    for i in range(T):
        inp=INP()
        #show(inp)
        a1=naive(*inp)
        a2=solve(*inp)
        if a1!=a2:
            print(inp)
            n,x=inp
            a,b=bin(n)[2:],bin(x)[2:]
            show(a,b)
            print('naive',a1)
            print('solve',a2)
            err+=1
        case+=1
    print('Tested',case,'case with',err,'errors')

mo=998244353
#mo=10**9+7

class Comb:
    def __init__(self,n=0,mo=10**9+7):
        self.size=1
        self.mo=mo
        self.fac=[0]*self.size
        self.inv=[1]*self.size
        self.fac[0]=1
        self.extend(n+1)
        return
    
    def extend(self,mx):
        x=max(2*self.size,mx)-self.size+1
        self.fac+=[0]*x
        self.inv+=[1]*x
        for i in range(self.size,self.size+x):
            self.fac[i]=i*self.fac[i-1]%self.mo
            self.inv[-1]*=i
            self.inv[-1]%=self.mo
        self.inv[-1]=pow(self.inv[-1],self.mo-2,self.mo)
        self.inv[-1]*=self.inv[self.size-1]
        self.inv[-1]%=self.mo
        for i in range(x+self.size-1,self.size,-1):
            self.inv[i-1]=self.inv[i]*(i)%self.mo
        self.size+=x
        return
    
    def fact(self,n):
        if n>=self.size:
            self.extend(n)
        return self.fac[n]
        
    def invf(self,n):
        if n>=self.size:
            self.extend(n)
        return self.inv[n]

    def comb(self,x,y):
        if y<0 or y>x:
            return 0
        if x>=self.size:
            self.extend(x)
        return self.fac[x]*self.inv[x-y]*self.inv[y]%self.mo

    def rcomb(self,x,y):
        if y<0 or y>x:
            return 0
        if x>=self.size:
            self.extend(x)
        return self.inv[x]*self.fac[x-y]*self.fac[y]%self.mo

    def cat(self,x):
        if x<0:
            return 0
        if 2*x>self.size:
            self.extend(x)
        return self.fac[2*x]*self.inv[x]*self.inv[x+1]%self.mo


show_flg=False
show_flg=True


cm=Comb(2,mo=998244353)

ans=0

D=2*10**5+2
#D=10
n,k=LI()
l=[0]*D
r=[0]*D
for i in range(n):
    a,b=LI()
    l[a]+=1
    r[b]+=1

c=0
ans=1
for i in range(D):
    if r[i]>0:
        if c>k:
            ans=0
        else:
            ans*=cm.comb(k-c+r[i],r[i])
            ans%=mo
    
    c-=r[i]    
    c+=l[i]

print((pow(k,n,mo)-ans)%mo)

yukicoder

結果

テストケース

ソースコード