思路:
首先用vector< node* > nodes记录每一行的文档内容(也记录了这一行的下标);
然后stack< node* > parents来找出上一层次的行。
vector< int > ans;//记录结果
vector< string > sel;//存储的是需要选择的标签或属性(可能是1个,可能是2个)
然后check一下sel存储那些选择内容是否符合当前行,如果sel.size=2的话,还要处理符合的当前行的上一层(父亲)是否符合sel[0](A)的选择。
最后输出ans数组即可。
代码:
#include <iostream>
#include <fstream>
#include <vector>
#include <stack>
#include <sstream>
using namespace std;
struct node{
string name,tag;
int level;
node* parent;
node(string n,string t,int l):name(n),tag(t),level(l),parent(0) {}
};
void divide(const string& line,vector<string>& 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)
{
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;
while(!parents.empty()) parents.pop();
string line;
getline(cin,line);
while(n--)
{
getline(cin,line);
int level=0;
while(line[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)
{
parents.pop();
}
now->parent=p;
//cout<<now->name<<"'s parent is "<<now->parent->name<<endl;
}
parents.push(now);
nodes.push_back(now);
}
vector<int> ans;
vector<string> sel;
while(m--)
{
getline(cin,line);
divide(line,sel);
ans.clear();
for(int i=0;i<nodes.size();i++)
{
int sl=sel.size()-1;
if(check(nodes[i],sel[sl]))
{
node* t=nodes[i]->parent;
sl--;
while(t&&sl>=0)
{
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;
}