CSP-元素选择器

题目

思路

在一开始做这题的时候我直接构造了棵树来存储题目给出的html文档，但是后面遇到了各种奇奇怪怪的bug，感觉修不过来了，就重新构思了一遍。然后发现其实不需要用树来存储，直接数组存储就行了。思路如下：用数组记录下输入的html文档，用vector记录选择器。非后代选择器的查找，只要遍历一边数组，判断其属性是否符合条件，符合则将其行号压入结果vector。对于后代选择器，则是在上面的基础上增加了一个步骤：判断其它的信息是否满足。详细来说就是在元素数组中，从被选择元素开始往前，与vector中元素一一对照，直到两边有一边元素对照完毕，若是vector中元素对照完毕则说明满足条件，否则不满足。将满足条件的行号压入结果vector。最后输出结果vector中节点数目及结果vector即可。注：这一题字符处理较为麻烦，还有要注意label的小写化。

代码

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

vector<string> qus;
vector<int> ans;

struct element
{
	string lable;
	string id;
	int level;
};

element e[105];

bool judge(int x)
{  
    int a=e[x].level,q=qus.size()-2;   
    for(int i=x-1;i>=0;i--)
    { 
        if(e[i].level<a)
        {
            if(qus[q]==e[i].lable||qus[q]==e[i].id)
                q--;       
            if(q<0) 
				return true;    
        }
    }
    return false;
}


int main()
{
	int n,m;
	string s,tmp;  
	cin>>n>>m;
	getchar();
	for(int i=1;i<=n;i++)
	{
        int plabel=-1;int pid=-1;int cnt=0;
		getline(cin,s);   
		for(int j=0;j<s.length();j++)
        {
            if(s[j]=='.')
            {
            	cnt++;
				continue;  	
			}    
			if(plabel==-1&&s[j]!='#')
               	plabel=j;   
            else 
				if(s[j]=='#')
             	  	pid=j;   
        }
        e[i].level=cnt/2;   
        if(pid==-1)
            e[i].lable=s.substr(plabel);
        else
        {  
            e[i].lable=s.substr(plabel,pid-plabel-1);
            e[i].id=s.substr(pid);
        }
        transform(e[i].lable.begin(), e[i].lable.end(), e[i].lable.begin(),::tolower);
	}

    for(int i=0;i<m;i++)
    {
        ans.clear();
        qus.clear();  
        getline(cin,s);
        stringstream ss(s);
        while(ss>>tmp)
        {
        	if(tmp[0]!='#') 
				transform(tmp.begin(), tmp.end(), tmp.begin(),::tolower);
       		qus.push_back(tmp);
		}
        int sz=qus.size();   
        for(int j=1;j<=n;j++)
        {  
            if(qus[sz-1][0]=='#'&&qus[sz-1]==e[j].id||qus[sz-1]==e[j].lable)
            {  
                if(sz==1) 
					ans.push_back(j);
                else 
					if(judge(j)) 
						ans.push_back(j);
            }
        }
        
        cout<<ans.size();
        for(int j=0;j<ans.size();j++)
        {
            cout<<" "<<ans[j];   
        }
        cout<<"\n";
    }
}