csp-元素选择器

题目

思路

这一题可以不用树结构解题。
对给定的结构化文档的每一行用结构体表示，结构体中含有标签、id以及级别。
对每一行进行遍历，初始化相应行的结构体数值。对给定的选择器依据空格得到字符段，并进行区分，如果不是后代选择器，则遍历给定的结构体文档输出结果。对于后代选择器，调用isMatch函数，从满足条件的第一行向上遍历，看是否满足选择器的要求，有则输出。

代码

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

const int maxn=120;
string str[maxn];

struct row
{
	string lab;//标签 
	string id;//id
	int lev;//级别 
}rows[maxn];

//vector<string> res;
vector<string> qus;//保存查询
vector<int> ans;//保存答案

void xiaoxie(string s)//变成小写
{
	for(int i=0;i<s.length();i++)
		if(s[i]>='A'&&s[i]<='Z')
			s[i]=s[i]+32;		
}
bool isMatch(int x)//判断当前后代查询能否查询到 
{  //向上一级查找
    int t=rows[x].lev;
	int q=qus.size()-2;
    for(int j=x-1;j>=0;j--)
    { //向上
        if(rows[j].lev==t-1)//是上一级别 
        {
            if(qus[q][0]!='#'&&qus[q]==rows[j].lab)q--;//标签配对
            else if(qus[q][0]=='#'&&qus[q]==rows[j].id)q--;//id配对
            t=rows[j].lev;
            if(q<0) return true;//都满足 
        }
    }
    return false;
}



int main()
{

	int n,m;
	int cnt,mark;
	string str,tmp;  
	cin>>n>>m;
	getchar();//吸收换行符 
	int tmp1,tmp2;//标记 
	for(int i=1;i<=n;i++)
	{
		getline(cin,str); 
        tmp1=-1;
		tmp2=-1;
		cnt=0;
		
		int len=str.length();
		for(int j=0;j<len;j++)
        {
            if(str[j]=='.')
                cnt++; 
            else if(str[j]=='#')
                tmp2=j;   //id
            else if(tmp1==-1&&str[j]!='#')
                tmp1=j;   //标签
            
        }
        
        rows[i].lev=cnt/2; //得到级别 
        if(tmp2==-1)//没有id 
        {  
        	rows[i].id="";
            rows[i].lab=str.substr(tmp1);//从tmp1开始的子串 
              
        }
        else
        {  
            rows[i].lab=str.substr(tmp1,tmp2-tmp1-1);//从tmp1开始，长度为 tmp2-tmp1-1
            rows[i].id=str.substr(tmp2);
        }
        xiaoxie(rows[i].lab);  //标签变成小写，便于比较 
	}

    //for(int i=1;i<=m;i++)
    while(m--)
    {
        ans.clear();
        qus.clear(); 
        getline(cin,str);
        for(int j=0;j<str.length();)//得到命令 
        {   
            mark=j;
            while(++mark<str.length()&&str[mark]!=' ');//找到空格 
            
            tmp=str.substr(j,mark-j);//空格前的一段字符 
            
            if(tmp[0]!='#')xiaoxie(tmp);//没有#，不是id 
            qus.push_back(tmp);
            j=mark+1;
        }
        int se=qus.size();
        
        for(int j=1;j<=n;j++)
        {  
            if(qus[se-1][0]=='#'&&qus[se-1]==rows[j].id||qus[se-1][0]!='#'&&qus[se-1]==rows[j].lab)
            {  //如果是当前要查询的行 
                if(se==1) ans.push_back(j); //单层 
                else if(isMatch(j))ans.push_back(j);//后代 
            }
        }
		cout<<ans.size();
        for(int j=0;j<ans.size();j++)
            cout<<" "<<ans[j];
        cout<<endl;

    }
	return 0;
}