Editing 0147 - expresie1

La ora de matematică Georgică a învăţat să evalueze expresii aritmetice. Pentru a verifica acest lucru profesorul de informatică îi propune lui Georgică să evalueze expresii aritmetice de forma:

<code>A  o1  B  o2  C</code>

Unde <code>A</code>, <code>B</code>, <code>C</code> sunt operanzi, iar <code>o1</code> şi <code>o2</code> sunt operatori.

Valorile permise operanzilor pot fi numai termeni din şirul <code>x1</code>, <code>x2</code>, …, <code>xn</code>, iar operatorii pot fi numai caracterele <code>+</code> (adunare) şi <code>*</code> (înmulţire).

Profesorul îi pune la dispoziție lui Georgică o valoare numerică <code>V</code> şi îi cere să determine valorile operanzilor <code>A</code>, <code>B</code> şi <code>C</code> pentru care expresia are valoarea <code>V</code>.

= Cerinţă =
Scrieţi un program, care pentru <code>V</code>, <code>n</code>, <code>x1</code>, <code>x2</code>, …, <code>xn</code> şi <code>o1</code>, <code>o2</code> date, determină indicii corespunzători termenilor din şirul <code>x1</code>, <code>x2</code>, …, <code>xn</code> ce corespund valorilor operanzilor <code>A</code>, <code>B</code> şi <code>C</code> pentru ca expresia dată să aibă valoarea <code>V</code>.

= Date de intrare =
Fişierul de intrare <code>input.txt</code> conţine pe prima linie numerele naturale <code>V</code>, <code>n</code>, <code>x1</code>, <code>x2</code>, …, <code>xn</code> separate între ele printr-un spaţiu şi pe linia a doua exact două caractere asociate operatorilor <code>o1</code> şi <code>o2</code>.

= Date de ieşire =
Fişierul de ieşire <code>output.txt</code> va conţine, pe prima linie separate între ele exact trei indici ai şirului <code>x1</code>, <code>x2</code>, …, <code>xn</code> corespunzători termenilor folosiţi pentru <code>A</code>, <code>B</code>, <code>C</code>. Dacă nu există soluţie se va scrie în fişier cifra <code>0</code> de trei ori separată prin câte un spaţiu: <code>0 0 0</code>.

= Restricţii şi precizări =

* <code>0 < n < 10001</code>

== Exemplul 1 ==
input.txt:

100 5 40 13 2 11 20

+*

output.txt:

5 3 1

Explicație:

<code>A = x5</code>, <code>B = x3</code>, <code>C = x1</code>

Expresia devine <code>20 + 2 * 40 = 100</code>

== Exemplul 2 ==
input.txt:

100 9999999999999 40 13 2 11 20

+*

Output:

Input-ul nu convine conditiilor

== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
def verificare(n):
    if not(1<=n<=10000):
        print("Input-ul nu convine conditiilor")
        exit()

def cit():
    global V, n, f, o1, o2
    with open("input.txt", "r") as fin:
        l=list(map(int, fin.readline().split()))
        V = l[0]
        n = l[1]
        verificare(n)
        f = [0] * 1005
        for i in range(1, n + 1):
            k = l[i+1]
            f[k] = i
        l=fin.readline()
        o1 = l[0]
        o2 = l[1]

def evalexp(A, B):
    global V, o1, o2
    if o1 == '+':
        if o2 == '+':
            t = V - (A + B)
        else:
            t = (V - A) / B
    else:
        if o2 == '+':
            t = V - A * B
        else:
            t = V / (A * B)
    if t > 0 and t == int(t):
        return int(t)
    return -1

def rezolva():
    global f
    for iA in range(1001):
        if f[iA] > 0:
            for iB in range(1001):
                if f[iB] > 0:
                    iC = evalexp(iA, iB)
                    if 0 <= iC < 1001 and f[iC] > 0:
                        with open("output.txt", "w") as fout:
                            fout.write(f"{f[iA]} {f[iB]} {f[iC]}")
                        return 1
    return 0

def main():
    cit()
    if rezolva() == 0:
        with open("output.txt", "w") as fout:
            fout.write("0 0 0")

if __name__ == "__main__":
    main()

</syntaxhighlight>