0673 - Dif Sub: Difference between revisions

From Bitnami MediaWiki
VisualWikitext
Pagină nouă: == Cerința == Considerăm un arbore binar cu n noduri în care fiecare nod este numerotat de la 1 la n și conține o valoare număr natural. Să se determine diferența în valoare absolută a sumei valorilor memorate în subarborele stâng al rădăcinii și suma valorilor memorate în subarborele drept al rădăcinii. == Date de intrare == Fișierul de intrare difsubin.txt conține pe prima linie numărul n. Fiecare dintre următoarele n linii contine câte 3 numere X st...
 
No edit summary
 
Line 1: Line 1:
== Cerința ==
= Cerința =
Considerăm un arbore binar cu n noduri în care fiecare nod este numerotat de la 1 la n și conține o valoare număr natural. Să se determine diferența în valoare absolută a sumei valorilor memorate în subarborele stâng al rădăcinii și suma valorilor memorate în subarborele drept al rădăcinii.
Considerăm un arbore binar cu <code>n</code> noduri în care fiecare nod este numerotat de la <code>1</code> la <code>n</code> și conține o valoare număr natural. Să se determine diferența în valoare absolută a sumei valorilor memorate în subarborele stâng al rădăcinii și suma valorilor memorate în subarborele drept al rădăcinii.
== Date de intrare ==
 
Fișierul de intrare difsubin.txt conține pe prima linie numărul n. Fiecare dintre următoarele n linii contine câte 3 numere X st dr; linia i + 1 din fișier conține informatiile despre nodul numerotat cu i: X reprezintă valoare din nod, st reprezintă numărul de ordine al descendentului stâng sau 0 dacă nodul i nu are descendent stâng, iar dr reprezintă numărul de ordine al descendentului drept sau 0 dacă nodul i nu are descendent drept.
= Date de intrare =
== Date de ieșire ==  
Fișierul de intrare <code>difsubIN.txt</code> conține pe prima linie numărul <code>n</code>. Fiecare dintre următoarele <code>n</code> linii contine câte <code>3</code> numere <code>X st dr</code>; linia <code>i + 1</code> din fișier conține informatiile despre nodul numerotat cu <code>i</code>: <code>X</code> reprezintă valoare din nod, <code>st</code> reprezintă numărul de ordine al descendentului stâng sau <code>0</code> dacă nodul <code>i</code> nu are descendent stâng, iar <code>dr</code> reprezintă numărul de ordine al descendentului drept sau <code>0</code> dacă nodul <code>i</code> nu are descendent drept.
Fișierul de ieșire difsubout.txt va conține pe prima linie numărul D, cu semnificația precizată.
 
== Restricții și precizări ==
= Date de ieșire =
*1 ≤ n ≤ 1000
Fișierul de ieșire <code>difsubOUT.txt</code> va conține pe prima linie numărul <code>D</code>, cu semnificația precizată. În cazul în care restricțiile nu sunt îndeplinite, se va afișa mesajul "Datele nu corespund restrictiilor impuse".
*valorile din nodurile arborelui vor fi mai mici sau egale cu 1.000.000
== Exemplu 1 ==
; difsubin.txt
: 6
: 2 3 5
: 6 0 6
: 1 0 0
: 7 1 2
: 4 0 0
: 10 0 0
; difsubout.txt
: 9
== Exemplu 2 ==
; difsubin.txt
: 5
: 3 2 3
: 7 4 0
: 1 0 0
: 5 0 0
: 2 0 0
; difsubout.txt
: 11
<br>
== Rezolvare ==
<syntaxhighlight lang="python" line>
#0673 - Dif Sub
class TreeNode:
    def __init__(self, value, left=None, right=None):
        self.value = value
        self.left = left
        self.right = right


def read_tree_from_file(file):
= Restricții și precizări =
    n = int(file.readline())
    if not (1 <= n <= 1000):
        return None


    nodes = [None]  # index 0 nu este folosit
* <code>1 ≤ n ≤ 1000</code>
* valorile din nodurile arborelui vor fi mai mici sau egale cu <code>1.000.000</code>


    for _ in range(n):
= Exemplul 1: =
        x, st, dr = map(int, file.readline().split())
<code>difsubIN.txt</code>
        if not (1 <= x <= 1000000):
6
            return None
2 3 5
        if not (0 <= st <= n):
6 0 6
            return None
1 0 0
        if not (0 <= dr <= n):
7 1 2
            return None
4 0 0
        nodes.append(TreeNode(x, st, dr))
10 0 0
<code>difsubOUT.txt</code>
9


    return nodes
= Exemplul 2: =
<code>difsubIN.txt</code>
1001
2 3 5
6 0 6
1 0 0
7 1 2
4 0 0
10 0 0
<code>difsubOUT.txt</code>
Datele nu corespund restrictiilor impuse


def calculate_subtree_sum(node, nodes):
== Rezolvare ==
     if node == 0:
<syntaxhighlight lang="python" line="1">
         return 0
def check_restrictions(n, x):
     return nodes[node].value + calculate_subtree_sum(nodes[node].left, nodes) + calculate_subtree_sum(nodes[node].right, nodes)
     if not (1 <= n <= 1000):
        return False
    if not all(0 <= val <= 1000000 for val in x):
         return False
     return True


def calculate_difference(root, nodes):
def SRD(k, st, dr, x):
     left_sum = calculate_subtree_sum(nodes[root].left, nodes)
     global d
     right_sum = calculate_subtree_sum(nodes[root].right, nodes)
    if st[k] != 0:
     return abs(left_sum - right_sum)
        SRD(st[k], st, dr, x)
     d += x[k]
     if dr[k] != 0:
        SRD(dr[k], st, dr, x)


def main():
def main():
     with open('difsubin.txt', 'r') as file:
    global d
         nodes = read_tree_from_file(file)
     with open("difsubIN.txt", "r") as f:
         n = int(f.readline().strip())
        x = [0] * (n + 1)
        st = [0] * (n + 1)
        dr = [0] * (n + 1)
        nod = [0] * (n + 1)


     if nodes is None:
        for i in range(1, n + 1):
         print("false")
            vals = f.readline().strip().split()
            if len(vals) != 3:
                with open("difsubOUT.txt", "w") as g:
                    g.write("Datele nu corespund restrictiilor impuse")
                return
 
            x[i] = int(vals[0])
            st[i] = int(vals[1])
            dr[i] = int(vals[2])
            if st[i]:
                nod[st[i]] = 1
            if dr[i]:
                nod[dr[i]] = 1
 
     if not check_restrictions(n, x[1:]):
         with open("difsubOUT.txt", "w") as g:
            g.write("Datele nu corespund restrictiilor impuse")
         return
         return


     root = 1 # Arborele începe de la rădăcină (nodul 1)
     rad = 1
     difference = calculate_difference(root, nodes)
     while nod[rad]:
        rad += 1


     with open('difsubout.txt', 'w') as file:
    d = 0
         file.write(str(difference))
    SRD(st[rad], st, dr, x)
    s = d
    d = 0
    SRD(dr[rad], st, dr, x)
 
     with open("difsubOUT.txt", "w") as g:
        if s > d:
            g.write(str(s - d))
         else:
            g.write(str(d - s))


if __name__ == "__main__":
if __name__ == "__main__":
Line 86: Line 103:


</syntaxhighlight>
</syntaxhighlight>
== Explicatie ==
Exemplul corespunde arborelui de mai jos, în care au fost marcate cu albastru valorile din noduri, iar cu roșu numerele de ordine ale nodurilor.
<br>
Suma valorilor nodurilor din subarborele stâng al rădăcinii este 7, iar suma valorilor nodurilor din subarborele drept al rădăcinii este 16. Diferența în valoare absolută a celor două valori este 9.

Latest revision as of 14:35, 18 May 2024

Cerința

Considerăm un arbore binar cu n noduri în care fiecare nod este numerotat de la 1 la n și conține o valoare număr natural. Să se determine diferența în valoare absolută a sumei valorilor memorate în subarborele stâng al rădăcinii și suma valorilor memorate în subarborele drept al rădăcinii.

Date de intrare

Fișierul de intrare difsubIN.txt conține pe prima linie numărul n. Fiecare dintre următoarele n linii contine câte 3 numere X st dr; linia i + 1 din fișier conține informatiile despre nodul numerotat cu i: X reprezintă valoare din nod, st reprezintă numărul de ordine al descendentului stâng sau 0 dacă nodul i nu are descendent stâng, iar dr reprezintă numărul de ordine al descendentului drept sau 0 dacă nodul i nu are descendent drept.

Date de ieșire

Fișierul de ieșire difsubOUT.txt va conține pe prima linie numărul D, cu semnificația precizată. În cazul în care restricțiile nu sunt îndeplinite, se va afișa mesajul "Datele nu corespund restrictiilor impuse".

Restricții și precizări

  • 1 ≤ n ≤ 1000
  • valorile din nodurile arborelui vor fi mai mici sau egale cu 1.000.000

Exemplul 1:

difsubIN.txt 

6
2 3 5
6 0 6
1 0 0
7 1 2
4 0 0
10 0 0

difsubOUT.txt 

9

Exemplul 2:

difsubIN.txt 

1001
2 3 5
6 0 6
1 0 0
7 1 2
4 0 0
10 0 0

difsubOUT.txt 

Datele nu corespund restrictiilor impuse

Rezolvare

<syntaxhighlight lang="python" line="1"> def check_restrictions(n, x): 

   if not (1 <= n <= 1000):
       return False
   if not all(0 <= val <= 1000000 for val in x):
       return False
   return True

def SRD(k, st, dr, x): 

   global d
   if st[k] != 0:
       SRD(st[k], st, dr, x)
   d += x[k]
   if dr[k] != 0:
       SRD(dr[k], st, dr, x)

def main(): 

   global d
   with open("difsubIN.txt", "r") as f:
       n = int(f.readline().strip())
       x = [0] * (n + 1)
       st = [0] * (n + 1)
       dr = [0] * (n + 1)
       nod = [0] * (n + 1)

       for i in range(1, n + 1):
           vals = f.readline().strip().split()
           if len(vals) != 3:
               with open("difsubOUT.txt", "w") as g:
                   g.write("Datele nu corespund restrictiilor impuse")
               return

           x[i] = int(vals[0])
           st[i] = int(vals[1])
           dr[i] = int(vals[2])
           if st[i]:
               nod[st[i]] = 1
           if dr[i]:
               nod[dr[i]] = 1

   if not check_restrictions(n, x[1:]):
       with open("difsubOUT.txt", "w") as g:
           g.write("Datele nu corespund restrictiilor impuse")
       return

   rad = 1
   while nod[rad]:
       rad += 1

   d = 0
   SRD(st[rad], st, dr, x)
   s = d
   d = 0
   SRD(dr[rad], st, dr, x)

   with open("difsubOUT.txt", "w") as g:
       if s > d:
           g.write(str(s - d))
       else:
           g.write(str(d - s))

if __name__ == "__main__": 

   main()

</syntaxhighlight>

Retrieved from "http://wiki.universitas.ro/index.php?title=0673_-_Dif_Sub&oldid=9804"