Editing 1715 - Inversiuni

Ludwig are o permutare <code>p=(p[1],p[2],...,p[N])</code> a mulțimii <code>{1,2,..,N}</code> și o masă pe care putea așeza numerele din permutare. Ludwig ia primul număr din permutare, adică <code>p[1]</code>, și îl așează pe masă. Al doilea număr, <code>p[2]</code>, îl pune fie în stânga lui <code>p[1]</code>, fie în dreapta lui <code>p[1]</code>. La fiecare pas, dacă s-au așezat pe masă deja numerele <code>p[1]</code>, <code>p[2]</code>, …, <code>p[i]</code>, atunci numărul <code>p[i+1]</code> este pus fie în stânga numerelor deja așezate, fie în dreapta lor.

= Cerința =
Ajutați-l pe Ludwig să determine o modalitate de așezare a întregii permutări pe masă astfel încât în final să se obțină o nouă permutare care are un număr minim de inversiuni.

= Date de intrare =
Fișierul de intrare <code>inversiuniIN.txt</code> conține pe prima linie numărul natural <code>N</code>, iar pe linia a doua, separate prin câte un spațiu, numerele <code>p[1]</code>, <code>p[2]</code>, …, <code>p[N]</code>.

= Date de ieșire =
Fișierul de ieșire <code>inversiuniOUT.txt</code> va conține un singur număr natural reprezentând numărul minim de inversiuni care se pot obține. În cazul în care restricțiile nu sunt îndeplinite, se va afișa mesajul "Datele nu corespund restrictiilor impuse".

= Restricții și precizări =

* <code>1 ≤ N ≤ 200 000</code>
* O inversiune în permutare este o pereche de indici <code>(i,j)</code> cu <code>i<j</code> și <code>p[i]>p[j]</code>. De exemplu, permutarea <code>p=(3,2,1,4)</code> are ca inversiune perechea de indici <code>(1,3)</code> pentru că <code>p[1]>p[3]</code>; o altă inversiune este perechea de indici <code>(2,3)</code> pentru că <code>p[2]>p[3]</code>. 

= Exemplul 1: =
<code>inversiuniIN.txt</code>
 4
 2 1 3 4
<code>inversiuniOUT.txt</code>
 0

=== Explicație ===
Se așează elementele pe masă astfel:

* <code>2</code>
* <code>1 2</code> (<code>1</code> este așezat la stânga)
* <code>1 2 3</code> (<code>3</code> este așezat la dreapta)
* <code>1 2 3 4</code> (<code>4</code> este așezat la dreapta)

Se obține permutarea identică, adică zero inversiuni.

== Exemplul 2: ==
<code>inversiuniIN.txt</code>
 0
 2 1 3 4
<code>inversiuniOUT.txt</code>
 Datele nu corespund restrictiilor impuse

== Rezolvare == 
<syntaxhighlight lang="python" line="1">
import sys

def verifica_restrictii(n):
    if not (1 <= n <= 200000):
        with open("inversiuniOUT.txt", "w") as fout:
            fout.write("Datele nu corespund restrictiilor impuse\n")
        sys.exit()

def update(a, n, x):
    i = x
    while i <= n:
        a[i] += 1
        i += (i & -i)

def query(a, x):
    s = 0
    i = x
    while i > 0:
        s += a[i]
        i -= (i & -i)
    return s

def main():
    with open("inversiuniIN.txt", "r") as fin:
        n = int(fin.readline().strip())
        verifica_restrictii(n)
        a = [0] * (n + 1)
        cnt = 0
        perm = list(map(int, fin.readline().strip().split()))
        for i in range(1, n + 1):
            x = perm[i - 1]
            mici = query(a, x)
            mari = i - mici - 1
            if mici < mari:
                cnt += mici
            else:
                cnt += mari
            update(a, n, x)

    with open("inversiuniOUT.txt", "w") as fout:
        fout.write(f"{cnt}\n")

if __name__ == "__main__":
    main()

</syntaxhighlight>