题解 | #数据分类处理#
数据分类处理
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;
}
}