HDU 1263 水果

map嵌套。特判了产地为空、产地不为空但水果为空的情况。

#include <bits/stdc++.h>
using namespace std;
int T,n,w;
string place,fruit;
map<string,map<string,int> >vis;
int main()
{
    ios::sync_with_stdio(false);
    cin>>T;
    while(T--)
    {
        cin>>n;
        vis.clear();
        for(int i=1;i<=n;i++)
        {
            cin>>fruit>>place>>w;
            if(!vis.count(place))//产地为空
            {
                map<string,int>tmp;
                tmp.clear();
                tmp.insert(pair<string,int>(fruit,w));
                vis.insert(pair<string,map<string,int> >(place,tmp));
            }
            else//产地不为空
            {
                if(!vis[place][fruit]) vis[place].insert(pair<string,int>(fruit,w));//水果为空
                else vis[place][fruit]+=w;
            }
        }
        for(auto i:vis)
        {
            string pl=i.first;
            printf("%s\n",pl.c_str());
            for(auto j:i.second)
            {
                string fr=j.first;
                int sumw=j.second;
                printf("   |----%s(%d)\n",fr.c_str(),sumw);
            }
        }
        if(T!=0)printf("\n");
    }
    return 0;
}

其实根本就不要特判产地或者水果为空,直接用下标访问,如果没有创建,它就会自动帮你创建。

for(int i=1;i<=n;i++)
{
    cin>>fruit>>place>>w;
    vis[place][fruit]+=w;
}

AC代码精简版:

#include <bits/stdc++.h>
using namespace std;
int T,n,w;
string place,fruit;
map<string,map<string,int> >vis;
int main()
{
    ios::sync_with_stdio(false);
    cin>>T;
    while(T--)
    {
        cin>>n;
        vis.clear();
        for(int i=1;i<=n;i++)
        {
            cin>>fruit>>place>>w;
            vis[place][fruit]+=w;
        }
        for(auto i:vis)
        {
            string pl=i.first;
            printf("%s\n",pl.c_str());
            for(auto j:i.second)
            {
                string fr=j.first;
                int sumw=j.second;
                printf("   |----%s(%d)\n",fr.c_str(),sumw);
            }
        }
        if(T!=0)printf("\n");
    }
    return 0;
}