Editing 3336 - acadele

== Cerinţa ==
Candyman are acadele de trei feluri: cu căpşuni, cu vişine şi cu zmeură, oricâte acadele din fiecare fel. Cei '''n''' copii de la grupa pregătitoare şi-au ales fiecare câte o acadea astfel încât cel mult doi copii şi-au ales cu vişine. Dacă notăm cu '''m''' numărul de moduri în care puteau să-şi aleagă fiecare câte o acadea, să se afle restul împărţirii lui '''m''' la '''2020'''.
== Date de intrare ==
Programul citește de la tastatură numărul '''n'''.
== Date de ieșire ==
Programul va afișa pe ecran restul împărţirii lui '''m''' la '''2020'''.
== Restricţii şi precizări ==
* '''1 &les; n &les; 1018'''

== Exemplul 1 ==
; Intrare
 2 
; Iesire
 9

== Exemplu 2 ==
; Intrare
 10000000000000000000
; Iesire
 Datele de intrare nu corespund restrictiilor impuse

== Rezolvare ==
<syntaxhighlight lang="python" line>
def compute_ways(n):
    mod = 2020
    if n == 1:
        return 3 % mod
    elif n == 2:
        return 9 % mod
    else:
        return (3 * 3 * pow(2, n-2, mod)) % mod


def main():
    n = int(input())

    if n > pow(10, 18):
        print("Datele de intrare nu corespund restrictiilor impuse")
        return

    ways = compute_ways(n)
    print("Datele de intrare corespund restrictiilor impuse")
    print(ways)


if __name__ == "__main__":
    main()
</syntaxhighlight>

== Explicatie ==
 Cei doi copii puteau să-şi aleagă acadelele astfel: (c,c), (z,z), (v,v), (c,v), (v,c), (c,z), (z,c), (v,z), (z,v).