1784 - Flori 3

Cerința

După o amorţeală care durase mai bine de 3 luni, Mama Natură îşi dădu seama că primăvara stătea să vină şi florile cu care trebuia curând să umple câmpiile nu erau încă pregătite. Înainte de începerea iernii, a lucrat să combine nuanţe din care să creeze flori pline de culoare, iar acum are camera plină de discuri de diferite culori, care aşteaptă să fie asamblate în flori viu colorate, fie pe post de mijloc, fie ca şi petale.

Pentru a forma o floare, Mama Natură alege un mijloc de orice culoare şi cel puţin K petale. Totodată, ea nu îşi doreşte să strice ordinea firească a lucrurilor şi de aceea nu va folosi niciodată două culori diferite de petale pentru aceeaşi floare. Ea admite, în schimb, flori cu petalele și mijlocul de aceeași culoare.

Deoarece timpul e scurt şi Mama Natură are lucruri mai importante de făcut decât să stea să asambleze flori, ea îşi cheamă în ajutor toate prietenele şi doreşte să îi dea fiecăreia ceva de lucru. Pentru aceasta, ea are la dispoziţie un şir D de N numere, unde numărul de pe poziţia i din șir reprezintă câte discuri de culoarea i a pregătit. Apoi, Mama Natură îşi pune M întrebări de forma x y, prin care doreşte să afle care este numărul maxim de flori care se pot forma folosind doar discuri de culori din intervalul [x, y] din şirul D.

Date de intrare

Fișierul de intrare flori3.in conține pe prima linie două numere naturale, N şi K, separate prin spaţiu, cu semnificaţia din enunţ. Pe următoarea linie se vor afla N numere naturale (elementele șirului D). Următoarea linie a fişierului va conţine numărul M şi va fi urmată de M linii conţinând perechi de numere întregi x și y, cu semnificaţia din enunţ.

Date de ieșire

Fișierul de ieșire flori3.out va conține, pe rânduri separate, M numere naturale, reprezentând răspunsul la fiecare dintre cele M întrebări.

Restricții și precizări

• 1 ≤ N ≤ 105
• 1 ≤ D[i] ≤ 109, Ɐ i = 1,..,N
• 1 ≤ M ≤ 105
• 1 ≤ K ≤ 1000
• 1 ≤ x ≤ y ≤ N

Exemplu:

flori3.in

4 4 
4 2 1 10 
2 
1 3 
1 4 

flori3.out

1 
3 

Explicație

Pentru prima întrebare, Mama Natură poate alege o singură floare, cu petale de culoarea 1 și mijloc de culoarea 2 sau 3.

Pentru a doua întrebare, putem forma 3 flori:

• 3 4 4 4 4
• 2 4 4 4 4 4 4
• 2 1 1 1 1

O alternativă: pentru a doua floare folosim mai puţine petale, rămânând in final două flori nefolosite de culoare 4.

Încărcare soluție

Lipește codul aici

import numpy as np


def main():
nmax = 1000006
SP = np.zeros(nmax, dtype=np.int64)
CP = np.zeros(nmax, dtype=np.int64)
sol = 0

with open("flori3.in", "r") as f:
    n, k = map(int, f.readline().split())
    for i in range(1, n + 1):
        x = int(f.readline())
        assert 1 <= x <= 1000000000
        SP[i] = SP[i - 1] + x // k
        CP[i] = CP[i - 1] + x % k
    m = int(f.readline())
    assert 1 <= n <= 100000
    assert 1 <= m <= 100000
    assert 1 <= k <= 1000
    for _ in range(m):
        x, y = map(int, f.readline().split())
        S = SP[y]
        C = CP[y]
        if x - 1 >= 0:
            S -= SP[x - 1]
            C -= CP[x - 1]
        if S <= C:
            sol = S
        else:
            x = (S - C) // (k + 1)
            sol = min(S - x, C + k * x)
            x += 1
            sol = max(sol, min(S - x, C + k * x))

with open("flori3.out", "w") as g:
    g.write(f"{sol}\n")

if __name__ == '__main__':
   main()