LeetCode 811. 子域名访问计数（C++）

一个网站域名，如"discuss.leetcode.com"，包含了多个子域名。作为顶级域名，常用的有"com"，下一级则有"leetcode.com"，最低的一级为"discuss.leetcode.com"。当我们访问域名"discuss.leetcode.com"时，也同时访问了其父域名"leetcode.com"以及顶级域名 "com"。

给定一个带访问次数和域名的组合，要求分别计算每个域名被访问的次数。其格式为访问次数+空格+地址，例如："9001 discuss.leetcode.com"。

接下来会给出一组访问次数和域名组合的列表cpdomains 。要求解析出所有域名的访问次数，输出格式和输入格式相同，不限定先后顺序。

示例 1:
输入: 
["9001 discuss.leetcode.com"]
输出: 
["9001 discuss.leetcode.com", "9001 leetcode.com", "9001 com"]
说明: 
例子中仅包含一个网站域名："discuss.leetcode.com"。按照前文假设，子域名"leetcode.com"和"com"都会被访问，所以它们都被访问了9001次。

示例 2
输入: 
["900 google.mail.com", "50 yahoo.com", "1 intel.mail.com", "5 wiki.org"]
输出: 
["901 mail.com","50 yahoo.com","900 google.mail.com","5 wiki.org","5 org","1 intel.mail.com","951 com"]
说明: 
按照假设，会访问"google.mail.com" 900次，"yahoo.com" 50次，"intel.mail.com" 1次，"wiki.org" 5次。
而对于父域名，会访问"mail.com" 900+1 = 901次，"com" 900 + 50 + 1 = 951次，和 "org" 5 次。

注意事项：

cpdomains 的长度小于 100。

每个域名的长度小于100。

每个域名地址包含一个或两个"."符号。

输入中任意一个域名的访问次数都小于10000。

C++

class Solution {
public:
    vector<string> subdomainVisits(vector<string>& cpdomains) 
    {
        vector<string> res;
        int n=cpdomains.size();
        map<string,int> tmp;
        for(int i=0;i<n;i++)
        {
            int idx0,idx1,idx2;
            int count=0;
            int m=cpdomains[i].size();
            for(int k=0;k<m;k++)
            {
                if(' '==cpdomains[i][k])
                {
                    idx0=k;
                }
                if(0==count)
                {
                    if('.'==cpdomains[i][k])
                    {
                        count++;
                        idx1=k;
                    }
                }
                else if(1==count)
                {
                    if('.'==cpdomains[i][k])
                    {
                        count++;
                        idx2=k;
                    }
                }
                else
                {
                    break;
                }
            }
            int num=stoi(cpdomains[i].substr(0,idx0));
            tmp[cpdomains[i].substr(idx1+1,m-idx1-1)]+=num;
            tmp[cpdomains[i].substr(idx0+1,m-idx0-1)]+=num;                
            if(2==count)
            {
                tmp[cpdomains[i].substr(idx2+1,m-idx2-1)]+=num;                
            }
        }
        for(auto it:tmp)
        {
            string a=to_string(it.second)+' ';
            a+=it.first;
            res.push_back(a);
        }
        return res;
    }
};