CSP201809-3 元素选择器

思路：

复杂模拟题

采用结构体存储元素信息，输入时，统一化为大写或小写。

考虑三种元素选择器
标签/id选择器:
直接遍历整个数组 分别与label/id作比较即可

*后代选择器:
①.首先遍历一遍元素 找出符合最后一个条件的所有元素 放入一个vector中
②.对vector中的每个元素
首先判断倒数第二个条件
从该元素的上一行开始 如果找到满足条件的行 并且level 满足a[k].level<levels 说明当前满足条件
此时要将levels更新为a[k].level

接着判断倒数第三个条件
从上一次停下的位置开始

 

#include<iostream>
#include<algorithm>
#include<math.h>
#include<string>
#include<string.h>
#include<stdio.h>
#include<vector>
#include<map>
#include<sstream>
using namespace std;
int n,m,tnt;
struct node
{
    string label;//标签 大小写不敏感 故可以全部转化为小写字母存储
    string id;//id属性
    int level;
}a[110]; 
string s;
//后代选择器用贪心的思路写 
//吧记录祖先 改为记录后代
//并且记录后代时 需要划分后代的等级 因为需要贪心
int main()
{
    cin>>n>>m;getchar();
    int N=n,M=m;
    while(N--)//将结构化文档初始读入 且记录前驱
    {
        tnt++;
        getline(cin,s);
        int levels=1,tmp=0;
        while(tmp<s.length())
        {
            if(s[tmp]!='.'||s[tmp+1]!='.')  break;
            tmp+=2;
            levels++;
        }
        a[tnt].level=levels;
        int i=1;
        int idx=s.find(' ');
        if(idx!=s.npos)//label 和 id
        {
            a[tnt].label=s.substr(tmp,idx-tmp);
            a[tnt].id=s.substr(idx+1,s.length()-(idx+1));
        }
        else a[tnt].label=s.substr(tmp,s.length()-tmp);

        transform(a[tnt].label.begin(),a[tnt].label.end(),a[tnt].label.begin(),::tolower);
    }

    while(M--)
    {
        getline(cin,s);
        int idx=s.rfind(' ',s.length()),last=s.length();
        if(idx==-1)//说明是标签选择器或id选择器
        {
            vector<int> ans;
            int tot=0;
            if(s[0]=='#')//id选择器
            {
                for(int i=1;i<=n;i++)
                {
                    if(a[i].id.compare(s)==0)
                    {
                        tot++;
                        ans.push_back(i);
                    }
                }
                cout<<tot<<" ";
                for(auto &entry:ans) cout<<entry<<" ";
                cout<<endl;
            }
            else//标签选择器
            {
                transform(s.begin(),s.end(),s.begin(),::tolower);
                for(int i=1;i<=n;i++)
                {
                    if(a[i].label.compare(s)==0)
                    {
                        tot++;
                        ans.push_back(i);
                    }
                }
                cout<<tot<<" ";
                for(auto &entry:ans) cout<<entry<<" ";
                cout<<endl;
            }
        }
        else//后代选择器
        {
            vector<int> ans;
            vector<string> limits;
            stringstream ss(s);
            while(ss>>s)
            {limits.push_back(s);}
            //先选择满足最后条件的
            string tmp=limits.back();
            if(tmp[0]=='#')
            {
                for(int i=1;i<=n;i++)
                    if(a[i].id.compare(tmp)==0)   ans.push_back(i);//可能是答案的 我们先记录下来
            }
            else
            {
                transform(tmp.begin(),tmp.end(),tmp.begin(),::tolower);
                for(int i=1;i<=n;i++)
                    if(a[i].label.compare(tmp)==0)   ans.push_back(i);//可能是答案的 我们先记录下来
            }
            
            int ans_tnt=ans.size();
            //处理前面的条件
            for(int i=0;i<ans.size();i++)
            {
                int idx=ans[i],levels=a[idx].level,now=idx-1;
                int tot=0;
                for(int j=limits.size()-2;j>=0;j--)
                {//从idx向前找 找到一个则更新初始的位置
                    for(int k=now;k>=0;k--)
                    {
                        if(limits[j][0]=='#')
                        {
                            if(a[k].id.compare(limits[j])==0&&a[k].level<levels)
                            {
                                tot++;
                                now=k-1;
                                levels=a[k].level;
                                break;
                            }
                        }
                        else
                        {
                            if(a[k].label.compare(limits[j])==0&&a[k].level<levels)
                            {
                                tot++;
                                now=k-1;
                                levels=a[k].level;
                                break;
                            }
                        }
                        
                    }
                }
                if(tot!=limits.size()-1) 
                {
                    ans[i]=-1;
                    ans_tnt--;
                }
            }
            cout<<ans_tnt<<" ";
            for(auto &entry:ans)
                if(entry!=-1)   cout<<entry<<" ";
            cout<<endl;
        }
        
    }
    return 0;
}