pat::a::a1004 命名空间参考

1004 Counting Leaves 更多...

类
struct	node

函数
int	main (istream &cin, ostream &cout)
	TEST (a1004, case1)

详细描述

1004 Counting Leaves

函数说明

main()

int pat::a::a1004::main	(	istream &	cin,
		ostream &	cout )

在文件 pat.cpp 第 4016 行定义.

                                                  {
                int n;
                int m;
                cin >> n >> m;
                if(n == 0) {
                    return 0;
                }
                unordered_map<string, node *> um;
                for(int i = 0; i < m; i++) {
                    string id;
                    int k;
                    cin >> id >> k;
                    if(!um.contains(id)) {
                        um[id] = new node(id);
                    }
                    auto *const nd = um[id];
                    for(int j = 0; j < k; j++) {
                        string cid;
                        cin >> cid;
                        if(!um.contains(cid)) {
                            um[cid] = new node(cid);
                        }
                        nd->children.insert(um[cid]);
                    }
                }
                queue<pair<node *, int>> q;
                q.push(make_pair(um["01"], 0));
                if(um["01"] == nullptr) {
                    cout << 1;
                    return 0;
                }
                int current_level = 0;
                int leaf_cnt      = 0;
                while(!q.empty()) {
                    auto [nd, level] = q.front();
                    q.pop();
                    if(level != current_level) {
                        if(current_level != 0) {
                            cout << ' ';
                        }
                        current_level = level;
                        cout << leaf_cnt;
                        leaf_cnt = 0;
                    }
                    if(nd->children.empty()) {
                        leaf_cnt++;
                    }
                    for(auto *c: nd->children) {
                        q.push(make_pair(c, level + 1));
                    }
                }
                cout << ' ' << leaf_cnt;
                return 0;
            }

被这些函数引用 TEST().

TEST()

pat::a::a1004::TEST	(	a1004	,
		case1	)

在文件 pat_test.cpp 第 1984 行定义.

                               {
                istringstream in("2 1\n"
                                 "01 1 02");
                auto out = ostringstream();
                main(in, out);
                ASSERT_EQ("0 1", out.str());
            }

引用了 main().