题解 | #数据分类处理#

数据分类处理

http://www.nowcoder.com/practice/9a763ed59c7243bd8ab706b2da52b7fd

利用 set 可以方便地对R数组进行排序和去重,set 是有序的,默认为升序排列,所以访问R中的元素时,直接auto elem:R即可。

#include <iostream>
#include <vector>
#include <set>

using namespace std;

int main() {
    int R, I, i, tmp;
    while(cin >> I) {
        vector<int> vi(I); // 存储 I 数组
        set<int> vr;    // 存储 R,去重+排序

        for (i = 0; i < I; i++) cin >> vi[i];
        cin >> R;
        for (i = 0; i < R; i++) {
            cin >> tmp;
            vr.insert(tmp);
        }

        int count = 0;
        vector<pair<int, vector<pair<int, int>>>> output; // (R[i], (idx, I[i]))

        for (const auto &elem : vr) {
            vector<pair<int, int>> contain_r;
            for (i = 0; i < I; i++) {
                // 遍历 I 数组,看是否包含 R[i]
                if (to_string(vi[i]).find(to_string(elem)) != string::npos) {
                    contain_r.push_back(make_pair(i, vi[i]));
                }
            }
            if (contain_r.size() > 0) { // 如果没有合适的I[i]
                output.push_back(make_pair(elem, contain_r));
                count += contain_r.size() * 2 + 2;
            }
        }

        cout << count << " "; // 先输出总数
        for (const auto &elem : output) {
            cout << elem.first << " " << elem.second.size() << " ";
            for (const auto &elem2 : elem.second) {
                cout << elem2.first << " " << elem2.second << " ";
            }
        }
        cout << endl;
    }
}

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务