2301 - Secv

Se consideră două numerele naturale K și S și un șir de N numere naturale a[1], a[2],…, a[N]. O secvenţă de lungime K este un subşir format din K elemente aflate pe poziţii consecutive în şir: a[i], a[i+1],.., a[i+k-1]. Parcurgând șirul de la stânga la dreapta, începând cu primul element, se elimină prima secvență de lungime K, cu suma elementelor strict mai mare decât numărul S. În urma ștergerii șirul va avea N-K elemente: a[1], a[2],…, a[N-K]. Operația de ștergere continuă după aceleași reguli până când nu mai există secvențe care pot fi eliminate.

Cerința

Să se scrie un program care citind numerele N, K, S și cele N elemente din șir rezolvă cerințele:

1) Determină numărul secvențelor care se vor elimina respectând condiția din enunț.

2) Considerând că în șirul citit nu sunt posibile eliminări de secvențe conform condiției din enunț, programul determină numărul de elemente a[i] din șir cu proprietatea următoare: ștergerea lui a[i] conduce la obținerea unui șir în care se mai poate elimina cel puțin o secvență de K elemente cu sumă strict mai mare ca S.

Date de intrare

Fișierul de intrare secv.in conține pe prima linie un număr natural P; numărul P poate avea doar valoarea 1 sau valoarea 2. A doua linie conține, în această ordine, separate prin câte un spațiu, numerele N, K și S. A treia linie conține, în ordine elementele șirului, despărțite prin câte un spațiu.

Date de ieșire

Dacă valoarea lui P este 1, se va rezolva numai cerinta 1). În acest caz, fişierul de ieşire secv.out va conține pe prima linie un număr natural reprezentând numărul secvențelor eliminate.

Dacă valoarea lui P este 2, se va rezolva numai cerinta 2). În acest caz, fişierul de ieşire secv.out va conține pe prima linie un număr natural reprezentând numărul elementelor din șir care au proprietatea că ștergerea fiecăruia în parte ar genera un șir în care se mai pot elimina cel puțin o secvență de K elemente cu sumă strict mai mare ca S.

Restricții și precizări

• 0 < N ≤ 1 000 000 și K ≤ N
• 0 < S ≤ 1 000 000 000
• 0 ≤ a[1], a[2],...,a[N] ≤ 1.000

Exemplul 1:

secv.in

1
14 3 7
1 2 1 3 1 4 5 2 1 4 1 8 2 3

secv.out

3

Încărcare soluție

Lipește codul aici

import sys

MAXN = 1000001
A = [0] * MAXN
B = [0] * MAXN
N = 0
K = 0
S = 0
P = 0
i = 0

def solve1():
    ul = 0
    Nr = 0
    global i
    for i in range(1, N + 1):
        B[ul + 1] = B[ul] + A[i]  # sume partiale
        ul += 1
        if ul >= K and B[ul] - B[ul - K] > S:
            Nr += 1
            ul -= K
    print(Nr)

def solve2():
    ul = 0
    global i
    for i in range(1, N + 1):
        B[ul + 1] = B[ul] + A[i]
        ul += 1
        if ul >= K and B[ul] - B[ul - K] > S:
            print(ul - K + 1, end=' ')
            ul -= K

def solve3():
    pr = 1
    ul = 0
    Nr = 0
    global i
    for i in range(1, N + 1):
        A[i] += A[i - 1]
    for i in range(1, N + 1):
        if i + K <= N:
            while pr <= ul and A[i + K] - A[i - 1] >= A[B[ul] + K] - A[B[ul] - 1]:
                ul -= 1
            B[ul + 1] = i
            ul += 1
        if A[B[pr] + K] - A[B[pr] - 1] - (A[i] - A[i - 1]) > S:
            Nr += 1
        if B[pr] == i - K:
            pr += 1
    print(Nr)

def main():
    sys.stdin = open('secv.in', 'r')
    sys.stdout = open('secv.out', 'w')
    global N, K, S, P, i
    P = int(input())
    N, K, S = map(int, input().split())
    A[1:N+1] = map(int, input().split())
    if P == 1:
        solve1()
    if P == 2:
        solve3()

if __name__ == "__main__":
    main()