CSP 201809-3 元素选择器

题目描述

想法

这是一道模拟题，首先是想办法存储整个文档，这里用一个pos结构体存储

struct pos
{
    string cla, tag;
    int spa,fa;
    pos() { spa = fa=0, cla = tag = " "; }
};

包括id和标签，fa表示自己的父亲索引，spa表示 " . " 的个数来表示层次。这里的父节点为逆序第一个spa值小于自身的节点索引值。
查找时，首先判断查找命令是否有空格，没有空格就代表标签或id选择器，进行一次遍历即可。
如果是后代选择器,我们已经存储的父亲信息所以我们这里从子节点向上查找，用一个vector存储路径，首相查找到孩子位置，再判断是否合法，判断条件：父节点的id或便签是否为vector中上一个路径值。
主要本题的标签值是不区分大小写的，所以这里我们统一用全下写来存储便签。

代码

#include <cstring>
#include <iostream>
#include<cstdio>
#include<string>
#include<vector>
using namespace std;
int T, n,m;
string str;
vector<string> qe;
struct pos
{
    string cla, tag;
    int spa,fa;
    pos() { spa = fa=0, cla = tag = " "; }
}p[102];
int ans[101];
 
string mystrlwr(string s)
{
    for (int i = 0; i < s.length(); i++)
        s[i] = tolower(s[i]);
    return s;
}

int main()
{ 
    scanf("%d%d", &n, &m);
    getchar();
    for (int i = 1; i <= n;i++)
    {
        getline(cin, str);
        int ii = 0;
        for (; ii < str.size()&&str[ii]=='.';ii++)
            ;
        p[i].spa = ii;
        str.erase(0, ii);
        ii = str.find(' ');
        p[i].tag = mystrlwr(str.substr(0, ii));
       // cout << "no" << i << " tag: " << p[i].tag << " ";
        if (ii != -1)
        {
            p[i].cla = str.substr(ii + 1);
        //    cout <<"cla :"<< p[i].cla << " ";
        }
        int j = i - 1;
        for (; j > 0&&p[j].spa>=p[i].spa;j--)
            ;
        p[i].fa = j; 
        // cout << endl;
    }
    
    while(m--)
    {
        getline(cin, str);
        if(str.find(' ')==-1)
        {
                int cnt = 0;
                if (str[0] != '#')
                    str = mystrlwr(str);
                for (int j=1; j <= n;j++)
                {
                    if (p[j].tag == str || p[j].cla == str)
                    {
                        ans[cnt++] = j;
                    }
                }
                printf("%d ",cnt);
                for (int j = 0; j < cnt;j++)
                    printf("%d ", ans[j]);
                printf("\n"); 
        }
        else
        {
            vector<string> path;
            vector<int> v;
            int pos = 0;
            while (pos != str.npos)
            {
                pos = str.find(' ');
                if (pos != str.npos)
                {
                    string tt;
                    tt = str.substr(0, pos);
                    if(tt[0]!='#')
                    tt=mystrlwr(tt);
                    path.push_back(tt);
                    str = str.substr(pos + 1, str.length() - (pos + 1));
                }
            }
            if(str[0]!='#')
                str = mystrlwr(str);
            path.push_back(str); //str就是所求 
                for (int i = 1; i <= n; i++)
                {
                    if (p[i].tag == str || p[i].cla == str)
                    {
                        int now = i;
                        bool ok = 1;
                        int j = path.size() - 2;
                        for (; j >= 0; j--)
                        {
                            if (p[now].fa == 0||(p[p[now].fa].cla != path[j]&&p[p[now].fa].tag != path[j]))
                            {
                                ok = 0;
                                break;
                            }
                            now = p[now].fa;
                        }
                        if (ok&&j==-1)
                            v.push_back(i);
                    }
                }
            printf("%d ", v.size());
            for (int cc = 0; cc <v.size(); cc++)
            {
                printf("%d ", v[cc]);
            }
            printf("\n");
 
        }
        
    } 
    return 0;
}