CSP元素选择器--100分题解

用到了栈的思想，注意大小写的区分与合并

#include <iostream>
#include <vector>
#include <stack>
#include <string>
#include <cstring>
#include <ctype.h>
#include <cstdio>
#include <algorithm>
#include <set>
using namespace std;
const int maxn=110;
int n1,m1;
int ind=0;//记录下标
set<int> res;//记录结果
struct node{
    int cnt=0;//记录点的个数
    string label="";
    string id="";
    vector<int> child;//指针域
}Node[maxn];
stack<int> sta;//栈
void split(string s,int num){
    int pos=-1;
    pos=s.find(" ");
    Node[ind].cnt=num;//点的个数
    if(pos!=-1)
    {
        int i=0;
        while(i<pos)//全部转化为小写字母label
        {
            s[i]=tolower(s[i]);
            i++;
        }
        Node[ind].label=s.substr(num,pos-num);
        Node[ind].id=s.substr(pos+1);
    }
    else
    {
        int i=0;
        while(s[i])//全部转化为小写字母label
        {
            s[i]=tolower(s[i]);
            i++;
        }
        Node[ind].label=s.substr(num);
    }
}
void add(string s){
    int num=0;//记录点的个数
    while(s[num]=='.'){
        num++;
    }
    while(!sta.empty())
    {
        int tp=sta.top();
        if(num>Node[tp].cnt)
        {
            //表明是孩子
            Node[tp].child.push_back(ind);
            break;
        }
        else {
            //不是孩子
            sta.pop();
        }
    }
    //找到了父节点或栈为空
    sta.push(ind);
    split(s,num);
    ind++;
}
void judge(string s,int ind)
{

    if(ind>=n1)return;
    int pos=-1;
    pos=s.find(" ");
    if(pos==-1)//只有一级
    {
        if(s[0]=='#')//id
        {
            if(Node[ind].id==s)res.insert(ind+1);
            for(int i=0;i<Node[ind].child.size();i++)
            {
                judge(s,Node[ind].child[i]);
            }

        }
        else//label
        {   int i=0;
            while(s[i])//全部转化为小写字母
            {
                s[i]=tolower(s[i]);
                i++;
            }
            if(Node[ind].label==s)res.insert(ind+1);
            for(int i=0;i<Node[ind].child.size();i++)
            {
                judge(s,Node[ind].child[i]);
            }
        }
    }
    else
    {
        string s1;
        s1=s.substr(0,pos);
        if(s1[0]!='#')
        {
            int i=0;
            while(s1[i])//全部转化为小写字母
            {
                s1[i]=tolower(s1[i]);
                i++;
            }
        }
        if(s1[0]=='#'&&Node[ind].id==s1)//父亲匹配id
        {
            for(int i=0;i<Node[ind].child.size();i++)
            {
              	judge(s.substr(pos+1),Node[ind].child[i]);
            }
        }
        else if(Node[ind].label==s1){
            //父亲匹配label
            for(int i=0;i<Node[ind].child.size();i++)
            {
                judge(s.substr(pos+1),Node[ind].child[i]);
            }
        }
        else judge(s,++ind);
    }
}
int main()
{
    int N,M;
    scanf("%d%d",&N,&M);
    n1=N;
    m1=M;
    getchar();
    string str;
    while(N--){
        getline(cin,str);
        add(str);
    }
    ind=0;
    while(M--)
    {
        getline(cin,str);
        res.clear();
        int i=0;
        judge(str,ind);
        printf("%d",res.size());
        for(set<int>::iterator it=res.begin();it!=res.end();it++)
        {
            printf(" %d",*it);
        }
        printf("\n");
    }
    return 0;
}