【CSP认证】201809_3 元素选择器

参考文章

【样例输入】

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

【样例输出】

3 6 9 11
1 6
0
2 9 11
1 11

代码实现

#include<iostream>
#include<cstring>
#include<sstream>
#include<vector>
#include<algorithm>
using namespace std;

string str; // 选择器 
string s;   // 查询 
int n,m;  
int cnt; // 记录标签/id的个数  

// 标签/id 结构体表示 
struct node{
	string name; // 名字 
	int row;     // 行数 
	int level;   // 层级 
}a[201];

// 字符串转小写 
string Lower(string s)
{
	string tmp = s;
	transform(tmp.begin(),tmp.end(),tmp.begin(),::tolower);
	return tmp;
}

// 输入处理 
void dealInput(string str,int r)
{
	if(str.find('.')==string::npos){ // 0级（只有标签） 
		a[cnt].name = Lower(str);
		a[cnt].row = r;
		a[cnt].level = 0; 
	}
	else{ // 1级及以上  
		int le = str.find_last_of('.')+1;  // .的个数
		string text,id;
		int pos = str.find(' '); // 查找空格 
		if(pos!=string::npos){ // 存在id 
			text = str.substr(le,pos-le); 
			id = str.substr(pos+1);
		} 
		else{ // 不存在id 
			text = str.substr(le);
		} 
		// 标签 
		a[cnt].name = Lower(text);
		a[cnt].row = r;
		a[cnt].level = le/2;
		// 存在id 
		if(!id.empty()){
			cnt++;
			a[cnt].name = id;
			a[cnt].row = r;
			a[cnt].level = le/2;
		}
	}
	cnt++;
} 

// 求某条查询的结果 
void dealQuery(vector<string> &v,vector<int> &ans)
{
	for(int i=0;i<cnt;i++){
		if(v.back()==a[i].name){ // 贪心：先找最后一个元素 
			int num = 1; // 匹配的个数 
			int begin = i-1; // 向上查找 
			int limit = a[i].level; // 层级
			for(int j=v.size()-2;j>=0;j--){ // 从倒数第二个开始 
				for(int k=begin;k>=0;k--){
					if(v[j]==a[k].name && a[k].level<limit){
						num++;
						begin = k-1;
						limit = a[k].level;
						break;
					}
				}
			}
				
			if(num==v.size()){ // 全部匹配 
				ans.push_back(a[i].row);
			} 
		}
	}			
	
}

int main()
{
	cin>>n>>m;
	
	getline(cin,str); // 吃掉前面的换行符 
	// 输入n条结构化文档的行数  
	for(int i=1;i<=n;i++){
		getline(cin,str);
		dealInput(str,i); // 保存每一个标签或id的信息 
	}
	
	// 输入m条待查询选择器的行数 
	while(m--){
		getline(cin,s); 
		
		vector<string> v; // 存储该行查询设计到的标签或id  
		istringstream ss(s);
		while(ss>>s){
			if(s[0]=='#'){ // id 
				v.push_back(s);
			}
			else{ // 标签 
				v.push_back(Lower(s));
			}
		}
		
		vector<int> ans; // 存储满足的行数 
		dealQuery(v,ans);
		cout<<ans.size(); // 结果大小 
		
		for(auto i:ans){
			cout<<" "<<i;
		}
		cout<<endl;	
	}
	
	return 0;
}