Editing 3575 - Br (section)

==Rezolvare==
===Rezolvare ver. 1===
<syntaxhighlight lang="python" line="1" start="1">


def validate_input(n, t, c_values, rounds):
    if n < 1 or n > 15000:
        raise ValueError("Invalid number of friends")
    if t < 1 or t > 10000:
        raise ValueError("Invalid number of rounds")
    if len(c_values) != n:
        raise ValueError("Invalid number of beer costs")
    if any(ci < 1 or ci > 100 for ci in c_values):
        raise ValueError("Invalid beer cost")
    for k, x in rounds:
        if k < 1 or k > n:
            raise ValueError("Invalid friend index")
        if x < 1 or x > 3000000:
            raise ValueError("Invalid amount of money")


def calculate_round_buying(n, c_values, k, x):
    beers_to_buy = [0] * n
    max_beers_bought = 0
    total_cost = 0

    for i in range(n):
        j = (k + i - 1) % n
        if total_cost + c_values[j] <= x:
            beers_to_buy[j] = 1
            total_cost += c_values[j]
            max_beers_bought += 1
        else:
            break

    if total_cost < x:
        for i in range(n):
            j = (k + i - 1) % n
            if beers_to_buy[j] == 0 and total_cost + c_values[j] <= x:
                beers_to_buy[j] = 1
                total_cost += c_values[j]
                max_beers_bought += 1
                if total_cost == x:
                    break
    return max_beers_bought


if __name__ == "__main__":
    try:
        with open("br.in") as f:
            n, t = map(int, f.readline().split())
            c_values = list(map(int, f.readline().split()))
            rounds = [tuple(map(int, line.split())) for line in f]
        print("Datele introduse corespund restricțiilor impuse.")
        validate_input(n, t, c_values, rounds)

        with open("br.out", "w") as f:
            for k, x in rounds:
                f.write(str(calculate_round_buying(n, c_values, k, x)) + "\n")
    except Exception as e:
        print("Datele introduse nu corespund restricțiilor impuse.")
</syntaxhighlight>