csp官网题目——元素选择器

元素选择器

思路

开始想用一个栈来进行存储以及判断，后来发现没有必要，数据也不大，我们直接顺序找所有点即可。
存储时存储.的个数，之后的判断层数就根据点的多少来直接判断。
对于一个查找，找到与最后一个相同的文档，从该位置开始，向前遍历，如果层级比当前低，则对这个进行一个判断，并且走到当前位置，判断标签或id是否相同。
还有一点是这个不是要每一级必须相邻，借用样例，加入要查找body h2，是可以找到答案的，尽管中间隔了一个div。
还有要注意的是要对标签进行全部小写的处理。

代码

#include<iostream>
#include<cstring>
#include<vector>
#include<sstream>
#include<string>
#include<algorithm>

using namespace std;

struct node
{
	string label,id;
	int count;
};

node a[105];

bool find(int& now,int& count,string s)
{
	for(int i=now;i>=1;i--)
		if(a[i].count<count)
		{
			now=i;
			count=a[i].count;
			if (a[i].label==s||a[i].id==s) 
				return true;
		}
	return false;
}

int main()
{
	int n,m,l1,l2,cou;
	string s;
	cin>>n>>m;
	getchar();
	for(int i=1;i<=n;i++)
	{
		getline(cin,s);
		l1=l2=-1; 	cou=0;
		for(int j=0;j<s.length();j++)
		{
			if (s[j]=='.')
				cou++;
			if (l1==-1&&s[j]!='#'&&s[j]!='.') 
				l1=j;
			if (s[j]=='#') 
				l2=j;
		}
		a[i].count=cou;
		if(l2==-1)
		{
			a[i].label=s.substr(l1);
			a[i].id="";
		}
		else 
		{
			a[i].label=s.substr(l1,l2-l1-1);
			a[i].id=s.substr(l2);
		}
		transform(a[i].label.begin(), a[i].label.end(), a[i].label.begin(), ::tolower);
	}
	
	vector<string> v;
	vector<int> ans;
	
	for (int i=1;i<=m;i++)
	{
		v.clear();
		ans.clear();
		getline(cin,s);
		int x=0,y=0;
		while (1)
		{
			string ss;
			x=s.find(" ",y);
			if (x==-1)
				ss=s.substr(y);
			else
				ss=s.substr(y,x-y); 
			if (ss[0]!='#')
				transform(ss.begin(),ss.end(),ss.begin(),::tolower);
			v.push_back(ss);
			if (x==-1) 
				break; 
			y=x+1;
		}

		for (int j=1;j<=n;j++)
			if (v[v.size()-1]==a[j].id||v[v.size()-1]==a[j].label)
			{
				int k,now,count;
				now=j; count=a[j].count;
				for (k=v.size()-2;k>=0;k--)
					if (!find(now,count,v[k]))
						break;
				if (k==-1) ans.push_back(j);
			}
		cout<<ans.size()<<" ";
		for(int j=0;j<ans.size();j++)
			cout<<ans[j]<<" ";
		cout<<endl;
	}
	
	return 0;
}