【python3】leetcode 811. Subdomain Visit Count（easy）

811. Subdomain Visit Count（easy）

A website domain like "discuss.leetcode.com" consists of various subdomains. At the top level, we have "com", at the next level, we have "leetcode.com", and at the lowest level, "discuss.leetcode.com". When we visit a domain like "discuss.leetcode.com", we will also visit the parent domains "leetcode.com" and "com" implicitly.

Now, call a "count-paired domain" to be a count (representing the number of visits this domain received), followed by a space, followed by the address. An example of a count-paired domain might be "9001 discuss.leetcode.com".

We are given a list cpdomains of count-paired domains. We would like a list of count-paired domains, (in the same format as the input, and in any order), that explicitly counts the number of visits to each subdomain.

Example 1:
Input: 
["9001 discuss.leetcode.com"]
Output: 
["9001 discuss.leetcode.com", "9001 leetcode.com", "9001 com"]
Explanation: 
We only have one website domain: "discuss.leetcode.com". As discussed above, the subdomain "leetcode.com" and "com" will also be visited. So they will all be visited 9001 times.

Example 2:
Input: 
["900 google.mail.com", "50 yahoo.com", "1 intel.mail.com", "5 wiki.org"]
Output: 
["901 mail.com","50 yahoo.com","900 google.mail.com","5 wiki.org","5 org","1 intel.mail.com","951 com"]
Explanation: 
We will visit "google.mail.com" 900 times, "yahoo.com" 50 times, "intel.mail.com" once and "wiki.org" 5 times. For the subdomains, we will visit "mail.com" 900 + 1 = 901 times, "com" 900 + 50 + 1 = 951 times, and "org" 5 times.

要求计算访问每一级域名的总数，题目说明了最多1个点或两个点，所以可以把顶级域名和二级域名用split分离出来

再使用字典把访问量加上

注意如果本身就是一个二级域名，那么web == second，要避免加两次所以需要判断

 class Solution:
    def subdomainVisits(self, cpdomains):
        """
        :type cpdomains: List[str]
        :rtype: List[str]
        """
        sub = []
        top = set()
        second = set()
        dic = {}
        for i in range(len(cpdomains)):
            nums , web = cpdomains[i].split()
            nums = int(nums)
            top = web.split('.')[-1]
            second = web.split('.')[-2]+'.'+web.split('.')[-1]
            dic[top] = dic.setdefault(top,0) + nums
            if second == web: dic[web] = dic.setdefault(web,0) + nums
            else:
                dic[second] = dic.setdefault(second,0) + nums
                dic[web] = dic.setdefault(web,0) + nums
        return [str(num)+' '+str(dm) for dm,num in dic.items()]

分离top和second的写法可以合并

class Solution:
    def subdomainVisits(self, cpdomains):
        """
        :type cpdomains: List[str]
        :rtype: List[str]
        """
        sub = []
        top = set()
        second = set()
        dic = {}
        for i in range(len(cpdomains)):
            nums , web = cpdomains[i].split()
            nums = int(nums)
            domains = web.split('.')
            for j in range(len(domains)):
                dic[".".join(domains[j:])] = dic.setdefault(".".join(domains[j:]),0) + nums
            
        return [str(num)+' '+str(dm) for dm,num in dic.items()]