元素选择器
Example
- input
11 5
html
..head
....title
..body
....h1
....p #subtitle
....div #main
......h2
......p #one
......div
........p #two
p
#subtitle
h3
div p
div div p
- output
3 6 9 11
1 6
0
2 9 11
1 11
解题思路
使用Node存储节点元素
查询分为三种情况:id选择器,文档中不同的元素不会有相同的id属性。标签选择器,匹配的标签可能有很多个。后代选择器,除最后一级外,前面的部分都可以尽量匹配层级小的元素。
我们可以从后代选择器的最后一个开始,在文档中寻找匹配的元素。如果能找到元素e,则在e的祖先元素中,寻找与该选择器的前一个选择器相匹配的元素。当遍历到根结点或者已经找到与后代选择器相匹配的元素时结束。
注意:后代选择器,只要有祖先元素满足选择器A即可,父元素并不一定要满足选择器A。
代码实现
#include <iostream>
#include <cstdio>
#include <string>
#include <sstream>
#include <vector>
using namespace std;
int cnt, idx;
struct Node {
int lev;
string label, id;
};
string trans(string tag) {
for (int i = 0; i < tag.size(); i++)
if ('A' <= tag[i] && tag[i] <= 'Z') tag[i] += 32;
return tag;
}
int main() {
int n, m;
cin >> n >> m;
getchar();
vector<Node> doc;
string line;
while (n--) {
getline(cin, line);
Node e;
cnt = 0;
while (line[cnt] == '.') cnt++;
e.lev = cnt / 2;
idx = line.find(" ");
if (idx != -1) {
e.label = trans(line.substr(cnt, idx - cnt));
e.id = line.substr(idx + 1, line.size() - idx - 1);
}
else {
e.label = trans(line.substr(cnt));
e.id = "";
}
doc.push_back(e);
}
while (m--) {
getline(cin, line);
vector<int> pos;
if (line.find(" ") == -1) {
for (int i = 0; i < doc.size(); i++) { if (doc[i].id == line || doc[i].label == trans(line)) { pos.push_back(i + 1); } }
}
else {
stringstream ss;
ss << line;
vector<string> selct;
while (ss >> line) {
if (line[0] != '#') line = trans(line);
selct.push_back(line);
}
string last = selct[selct.size() - 1];
for (int i = 0; i < doc.size(); i++) {
if (last != doc[i].id && last != doc[i].label) continue;
idx = selct.size() - 2;
int level = doc[i].lev;
for (int k = i; k >= 0 && idx >= 0; k--) {
Node e = doc[k];
string selector = selct[idx];
if (e.lev < level) {
level = e.lev;
if (selector == e.id || selector == e.label) idx--;
}
}
if (idx < 0) pos.push_back(i + 1);
}
}
cout << pos.size();
for (int i = 0; i < pos.size(); i++) {
cout << ' ' << pos[i];
}
cout << endl;
}
return 0;
}