HackerRank XML 1 – Find the Score solution in Python

HackerRank XML 1 – Find the Score solution in Python – You are given a valid XML document, and you have to print its score. The score is calculated by the sum of the scores of each element. For any element, the score is equal to the number of attributes it has.

Input Format

The first line contains N, the number of lines in the XML document.
The next N lines follow containing the XML document.

Output Format

Output a single line, the integer score of the given XML document.

HackerRank XML 1 – Find the Score solution in Python 2.

n = int(raw_input())
xml = ""
for i in range(n):
    xml += raw_input()
import xml.etree.ElementTree as etree
tree = etree.ElementTree(etree.fromstring(xml))
score = 0
for elem in tree.iter():
	score += len(elem.attrib)
print score

XML 1 – Find the score solution in Python 3.

N = int(input())

def count_of_attr(root):
    count = len(root.attrib)
    for child in root:
        count += count_of_attr(child)
    return count

import xml.etree.ElementTree as etree
count = 0
xml = ''
for i in range(N):
    xml += input()

tree = etree.ElementTree(etree.fromstring(xml))
root = tree.getroot()
count = count_of_attr(root)
    
print(count)

Problem solution in pypy programming.

def get_attr_number(node):
    # your code goes here
    count = len(node.attrib)
    for x in node:
        count += get_attr_number(x)
    return count

Problem solution in pypy3 programming.

def get_attr_number(node):
    return len(node.attrib) + sum(get_attr_number(child) for child in node)