99022024-03-18 14:53:4042Szitakötő (50 pont)pypy3Futási hiba 36/50171ms167700 KiB
# O(NlogN)

from sys import stdin, stdout
input=stdin.readline

mod=10**9+7

def solv():
    N,K = [int(x) for x in input().split()]
    W = [int(x) for x in input().split()]
    if N==1:
        print(2)
        return
    if K==1:
        print(0)
        return
    preW=[0]*N
    for i in range(N):
        preW[i]=preW[i-1]+W[i]
    i=K-2
    while preW[i]*2 > preW[K-1]:
        i-=1
    # 2^(i+1) lehetoseg balra
    if K==N:
        print(pow(2,i+2,mod))
        return
    # K<N:
    d={}
    j=K
    while j<N and 2*preW[K-1] > preW[j]:
        j+=1
    if j==N:
        print(pow(2,i+1+N-K,mod))
        return
    if K==j:
        print(0)
        return
    d={K-1:j}
    for k in range(K,N):
        while j<N and 2*preW[k] > preW[j]:
            j+=1
        if j==N:
            break
        if k+1==j:
            print(0)
            return
        d[k]=j
    preW.append(0)
    W[N-1]=preW[N-1]=0
    W[N-2]=preW[N-2]=0
    for k in range(N-2,K-2,-1):
        if k not in d:
            W[k]=pow(2,N-k-1,mod)
            preW[k]=preW[k+1]+W[k]
        else:
            W[k]=preW[k+1]-preW[d[k]]
            preW[k]=preW[k+1]+W[k]
    print((pow(2,i+1,mod)*W[K-1])%mod)
    
solv()
RészfeladatÖsszpontTesztVerdiktIdőMemória
base36/50
1Elfogadva0/057ms87848 KiB
2Futási hiba0/0155ms167700 KiB
3Elfogadva1/148ms88524 KiB
4Elfogadva1/148ms88696 KiB
5Elfogadva1/143ms88768 KiB
6Elfogadva1/148ms89444 KiB
7Elfogadva1/143ms89768 KiB
8Elfogadva1/141ms90156 KiB
9Elfogadva1/146ms89900 KiB
10Elfogadva2/245ms90412 KiB
11Elfogadva2/250ms90068 KiB
12Elfogadva2/248ms90732 KiB
13Elfogadva2/252ms93828 KiB
14Elfogadva2/250ms93892 KiB
15Elfogadva2/252ms94056 KiB
16Elfogadva2/252ms93952 KiB
17Elfogadva2/254ms94320 KiB
18Elfogadva2/254ms94092 KiB
19Elfogadva2/252ms94256 KiB
20Elfogadva2/252ms94404 KiB
21Elfogadva1/148ms94364 KiB
22Futási hiba0/2149ms164604 KiB
23Futási hiba0/2150ms164368 KiB
24Futási hiba0/2171ms164628 KiB
25Futási hiba0/2165ms164548 KiB
26Elfogadva2/276ms117688 KiB
27Futási hiba0/2146ms164612 KiB
28Futási hiba0/2150ms164016 KiB
29Elfogadva2/2137ms135968 KiB
30Futási hiba0/2158ms164268 KiB
31Elfogadva2/275ms117048 KiB