csp2018-09-3_元素选择器

csp2018-09-3_元素选择器

---

题目描述

题目思路

这是一个大的模拟题，其中也包含了一些对于字符串的处理。首先我们定义了一个node类型结构体表示输入的每一层的标签，其中包含label标签，id和输入的时候标签前面的点的数目cnt。由于标签label对大小写不敏感，所以为了处理方便我将所有的label标签使用tolower函数转化为小写字符。之后query数组使用存储查询记录的，ans数组存储结果。对于标签选择器和id选择器处理很简单，关键是后代选择器中需要寻找查询到父亲，这里我们只需要找到比当前cnt小的第一个标签的cnt即可，这里是使用search函数实现查找的。

代码实现

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
const int N = 110;

int n,m;
string s;
vector<string> query;	//存储查询语句
vector<int> ans; 

struct node{
	string lable,id;	//标签和id
	int cnt;			//点的数量 
}nd[N];

void ToLower(string &s){
	for(int i = 0; i < s.length(); i++){
		s[i] = tolower(s[i]);
	}
}

bool search(int &r, int &cnt, string s){
	for(int i = r; i >= 1; i--){
		if(nd[i].cnt < cnt){
			cnt = nd[i].cnt,r = i;
			if(s == nd[i].lable || s == nd[i].id) return true;
		}
	}
	return false;
}

void input(){
	cin >> n >> m;
	getchar();
	for(int i = 1; i <= n; i++){
		getline(cin,s);
		int pos1,pos2,num;//标签的起始位置，id的起始位置，点的数量
		pos1 = pos2 = -1,num = 0;
		for(int j = 0; j < s.length(); j++){
			if(s[j] == '.') num++;
			else if(pos1 == -1 && s[j] != '#') pos1 = j;
			else if(s[j] == '#') pos2 = j;
		}
		nd[i].cnt = num;
		if(pos2 == -1){//不存在id属性
			nd[i].lable = s.substr(pos1);
			nd[i].id = "";
		}
		else{//存在id属性 
			nd[i].lable = s.substr(pos1,pos2 - pos1 - 1);
			nd[i].id = s.substr(pos2);
		}
		ToLower(nd[i].lable);
	}
}

int main(){
	input();
	
	for(int i = 0; i < m; i++){
		char c[100];
		gets(c);
		if(!query.empty()) query.clear();
		if(!ans.empty()) ans.clear();
		char *sp = strtok(c," ");
		while(sp){
			query.push_back(sp);
			sp = strtok(NULL," ");
		}
		int sz = query.size();
		for(int j = 0; j < sz; j++){
			if(query[j][0] != '#') ToLower(query[j]);
		}
		for(int j = 1; j <= n; j++){
			if(query[sz - 1] == nd[j].id || query[sz - 1] == nd[j].lable){
				int r = j,cnt = nd[j].cnt,k = sz - 2;
				for(; k >= 0; k--)
					if(!search(r,cnt,query[k])) break;
				
				if(k < 0)
					ans.push_back(j);
			}
		}
		cout << ans.size();
		for(int j = 0; j < ans.size(); j++){
			cout << " " << ans[j];
		}
		cout << endl;
	}
}