Solutions to Python XML section

Author: sknsht

Python/XML/XML 1 - Find the Score/Solution.py

@@ -0,0 +1,14 @@
+import sys
+import xml.etree.ElementTree as etree
+
+
+def get_attr_number(node):
+    return len(node.attrib) + sum(get_attr_number(child) for child in node);
+
+
+if __name__ == '__main__':
+    sys.stdin.readline()
+    xml = sys.stdin.read()
+    tree = etree.ElementTree(etree.fromstring(xml))
+    root = tree.getroot()
+    print(get_attr_number(root))

Python/XML/XML2 - Find the Maximum Depth/Solution.py

@@ -0,0 +1,22 @@
+import xml.etree.ElementTree as etree
+
+maxdepth = 0
+
+
+def depth(elem, level):
+    global maxdepth
+    if (level == maxdepth):
+        maxdepth += 1
+
+    for child in elem:
+        depth(child, level + 1)
+
+
+if __name__ == '__main__':
+    n = int(input())
+    xml = ""
+    for i in range(n):
+        xml = xml + input() + '\n'
+    tree = etree.ElementTree(etree.fromstring(xml))
+    depth(tree.getroot(), -1)
+    print(maxdepth)