Leetcode 811. Subdomain Visit Count

Tyan

发布于 2021-03-02 16:30:36

2320

发布于 2021-03-02 16:30:36

文章被收录于专栏：SnailTyan

文章作者：Tyan 博客：noahsnail.com | CSDN | 简书

1. Description

2. Solution

Version 1

class Solution:
    def subdomainVisits(self, cpdomains):
        stat = {}
        for cpdomain in cpdomains:
            data = cpdomain.split(' ')
            count = int(data[0])
            domain = data[1]
            self.subdomainStat(stat, domain, count)

        result = []
        for key, value in stat.items():
            string = str(value) + ' ' + key
            result.append(string)
        return result


    def subdomainStat(self, stat, domain, count):
        if domain in stat:
            stat[domain] += count
        else:
            stat[domain] = count

        subdomain = domain.split('.', 1)
        if len(subdomain) > 1:
            self.subdomainStat(stat, subdomain[1], count)

Version 2

class Solution:
    def subdomainVisits(self, cpdomains):
        stat = {}
        for cpdomain in cpdomains:
            data = cpdomain.split(' ')
            count = int(data[0])
            domain = data[1]
            while True:
                if domain in stat:
                    stat[domain] += count
                else:
                    stat[domain] = count
                subdomain = domain.split('.', 1)
                if len(subdomain) == 1:
                    break
                else:
                    domain = subdomain[1] 

        result = []
        for key, value in stat.items():
            string = str(value) + ' ' + key
            result.append(string)
        return result