csp模拟-元素选择器

题目

分析

这个模拟题需要记录的是结构化文档，结构化文档需要记录的有：标签、id属性、以及有层次的包含关系。使用struct记录：

struct node{
	string name,tag;//标签和属性
	int level;//层次
	node* parent;//上一个node
	node(string n,string t,int l){name=n;tag=t;level=l;parent=null}
}; 

然后是如何读入这个具有层次的结构化文档：
首先用vector< node* > nodes记录每一行的文档内容（也记录了这一行的下标）；
然后stack< node* > parents来找出上一层次的行
输入一行，读入name,tag；需要稍微处理的就是当前node的上一层次是哪个，这个时候，就从stack栈中寻找，栈是始终level从小到大排列的。找到stack_top->level小于当前level的，就是当前node的上一层。

最后处理每一个选择器的内容：
核心部分是check：

bool check(node* t,const string& s){
//要处理的哪一行node，以及选择器的内容s
	if(s[0]=='#') return s==t->tag;//如果是选择属性，返回这一行的属性是否和选择器的一样。
	//下面是判断这一行的标签是否和选择器的标签一样
	//这个需要特殊处理的原因是标签大小写不敏感
	if(s.size()!=t->name.size()) return false;
	for(int i=0;i<s.size();i++){
		if(tolower(s[i])!=tolower(t->name[i])) return false;
	}
	return true;
}

大概过程就是：
vector< int > ans;//记录结果
vector< string > sel;//存储的是需要选择的标签或属性（可能是1个，可能是2个）
然后check一下sel存储那些选择内容是否符合当前行，如果sel.size=2的话，还要处理符合的当前行的上一层（父亲）是否符合sel[0](A）的选择。

所以呢，这道题的核心就是如何存储这个结构化的文档，如何找到当前行的上一层行。

代码

#include <iostream>
#include <fstream>
#include <vector>
#include <stack>
#include <sstream>

using namespace std;

struct node{
	string name,tag;//标签和属性
	int level;//层次
	node* parent;//上一个node
	node(string n,string t,int l){name=n;tag=t;level=l;parent=null}
};

void divide(const string& line,vector<string>& sel){//把这一行选择器分离到sel数组里面
	sel.clear();
	string token;
	token.clear();
	for(int i=0;i<line.size();i++){
		if(line[i]==' '){
			sel.push_back(token);
			token.clear();
		}
		else token+=line[i];
	}
	sel.push_back(token);
}

bool check(node* t,const string& s){
//要处理的哪一行node，以及选择器的内容s
	if(s[0]=='#') return s==t->tag;//如果是选择属性，返回这一行的属性是否和选择器的一样。
	//下面是判断这一行的标签是否和选择器的标签一样
	//这个需要特殊处理的原因是标签大小写不敏感
	if(s.size()!=t->name.size()) return false;
	for(int i=0;i<s.size();i++){
		if(tolower(s[i])!=tolower(t->name[i])) return false;
	}
	return true;
}
int main(){
	int n,m;
	scanf("%d%d",&n,&m);
	
	vector<node*> nodes;//记录每一行的结构化文档
	nodes.clear();
	stack<node*> parents;//用来找出上一个层次的行（node）
	while(!parents.empty()) parents.pop();
	
	string line;
	getline(cin,line);
	while(n--)
	{//处理n行结构化文档
		getline(cin,line);
		int level=0;
		while(line[level]=='.') level++;//用'.'来划分层次的，level记录有几个'.'
		stringstream ss(line.substr(level));
		string name,tag;
		ss>>name>>tag;//读入标签和属性
		node* now=new node(name,tag,level);
		if(!parents.empty()){
			node* p;
			while(p=parents.top(),p->level>=level){//栈顶的level更大，层次更低，不是当前行的上一个层次，需要不断寻找
				parents.pop();
			}
			now->parent=p;
		}
		parents.push(now);//当前的行压栈
		nodes.push_back(now);//记录当前行
	} 
	
	vector<int> ans;//记录结果
	vector<string> sel;//存储的是需要选择的标签或属性（可能是1个，可能是2个）
	while(m--){//然后处理选择器
		getline(cin,line);//读入一行
		divide(line,sel);
		ans.clear();
		for(int i=0;i<nodes.size();i++){//循环遍历每一行（nodes）
			int sl=sel.size()-1;//sl=0/sl=1
			if(check(nodes[i],sel[sl])){
				node* t=nodes[i]->parent;//满足的node的parent
				sl--;
				while(t&&sl>=0){//处理的是操作3：后代选择
				//判断parent是否满足选择器中的A
					if(check(t,sel[sl])) sl--;
					t=t->parent;
				}
				if(sl==-1) ans.push_back(i+1);//如果符合，加入这一行的下标
			}
		}		
		printf("%d ",ans.size());
		for(int i=0;i<ans.size();i++) printf("%d ",ans[i]);
		if(m!=0) printf("\n");
	}
	return 0;
}