#562866 (PyPy3) No.1238 選抜クラス

提出ソース

結果

問題	No.1238 選抜クラス
ユーザー	Keroru
提出日時	2020-10-06 03:06:27
言語	PyPy3 (7.3.15)
結果	AC
実行時間	216 ms / 2,000 ms
コード長	4,660 bytes
コンパイル時間	412 ms
コンパイル使用メモリ	82,304 KB
実行使用メモリ	122,368 KB
最終ジャッジ日時	2024-07-19 22:46:06
合計ジャッジ時間	8,213 ms
ジャッジサーバーID （参考情報）	judge2 / judge3

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	141 ms 92,288 KB
testcase_01	AC	143 ms 92,416 KB
testcase_02	AC	150 ms 92,800 KB
testcase_03	AC	144 ms 92,032 KB
testcase_04	AC	143 ms 92,416 KB
testcase_05	AC	146 ms 92,288 KB
testcase_06	AC	144 ms 92,416 KB
testcase_07	AC	151 ms 92,800 KB
testcase_08	AC	151 ms 92,800 KB
testcase_09	AC	167 ms 92,800 KB
testcase_10	AC	158 ms 92,416 KB
testcase_11	AC	155 ms 92,672 KB
testcase_12	AC	145 ms 92,544 KB
testcase_13	AC	147 ms 92,288 KB
testcase_14	AC	145 ms 92,672 KB
testcase_15	AC	153 ms 92,672 KB
testcase_16	AC	151 ms 92,800 KB
testcase_17	AC	188 ms 121,984 KB
testcase_18	AC	191 ms 120,320 KB
testcase_19	AC	206 ms 120,832 KB
testcase_20	AC	206 ms 119,040 KB
testcase_21	AC	192 ms 117,376 KB
testcase_22	AC	187 ms 122,240 KB
testcase_23	AC	193 ms 121,984 KB
testcase_24	AC	196 ms 122,240 KB
testcase_25	AC	186 ms 122,112 KB
testcase_26	AC	201 ms 122,112 KB
testcase_27	AC	196 ms 122,368 KB
testcase_28	AC	216 ms 121,984 KB
testcase_29	AC	199 ms 120,832 KB
testcase_30	AC	160 ms 92,672 KB
testcase_31	AC	161 ms 92,544 KB
testcase_32	AC	176 ms 93,184 KB
testcase_33	AC	146 ms 92,672 KB
testcase_34	AC	195 ms 113,152 KB
testcase_35	AC	164 ms 92,416 KB
testcase_36	AC	159 ms 92,544 KB
testcase_37	AC	162 ms 92,544 KB
testcase_38	AC	191 ms 119,296 KB

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

ソースコード

raw source code

import sys,bisect,string,math,time,functools,random,fractions
from heapq import heappush,heappop,heapify
from collections import deque,defaultdict,Counter
from itertools import permutations,combinations,groupby
rep=range;R=range
def Golf():n,*t=map(int,open(0).read().split())
def I():return int(input())
def S_():return input()
def IS():return input().split()
def LS():return [i for i in input().split()]
def MI():return map(int,input().split())
def LI():return [int(i) for i in input().split()]
def LI_():return [int(i)-1 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 RA():return map(int,open(0).read().split())
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 INP():
    N=10;A=10
    n=random.randint(1,N)
    a=[random.randint(1,A) for i in range(n)]
    return n,a
def Rtest(T):
    case,err=0,0
    for i in range(T):
        inp=INP()
        a1=naive(*inp)
        a2=solve(*inp)
        if a1!=a2:
            print((a1,a2),inp)
            err+=1
        case+=1
    print('Tested',case,'case with',err,'errors')
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 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)

mo=10**9+7
#mo=998244353
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('LRUD',FourNb))
l_alp=string.ascii_lowercase
#sys.setrecursionlimit(10**9)
read=sys.stdin.buffer.read;readline=sys.stdin.buffer.readline;input=lambda:sys.stdin.readline().rstrip()

class UnionFind:
    def __init__(self, n):
        # 負  : 根であることを示す。絶対値はランクを示す
        # 非負: 根でないことを示す。値は親を示す
        self.table = [-1] * n
 
    def _root(self, x):
        if self.table[x] < 0:
            return x
        else:
            # 経路の圧縮
            self.table[x] = self._root(self.table[x])
            return self.table[x]
 
    def find(self, x, y):
        return self._root(x) == self._root(y)
 
    def union(self, x, y):
        r1 = self._root(x)
        r2 = self._root(y)
        if r1 == r2:
            return
        # ランクの取得
        d1 = self.table[r1]
        d2 = self.table[r2]
        if d1 <= d2:
            self.table[r2] = r1
            if d1 == d2:
                self.table[r1] -= 1
        else:
            self.table[r1] = r2
    
    def __str__(self):
        rt=[i if j<0 else j for i,j in enumerate(self.table)]
        return str(rt)

show_flg=False
show_flg=True


ans=0
n,k=LI()
a=sorted([i-k for i in LI()])

N=n*(100+1)

M=2*N+4
dp=[0]*M
#dp[0]=1


for i in range(n):
    nx=[0]*M
    c=a[i]
    nx[c]+=1
    for i in range(N+1):
        nx[i+c]+=dp[i]
        if i!=0:
            nx[-i+c]+=dp[-i]
        
    dp=[(dp[i]+nx[i])%mo for i in range(M)]
    #print(sum(dp),dp,nx)

ans=sum(dp[:N+1])
print(ans%mo)

yukicoder

結果

テストケース

ソースコード