CCF-CSP 201809-3 元素选择器【模拟】（100分）

试题编号： 201809-3
试题名称： 元素选择器
时间限制： 1.0s
内存限制： 256.0MB
问题描述：




写大模拟也太累了吧

代码如下（100分）：

#include <iostream>
#include <string>
#include <cctype>
#include <map>
#include <vector>
#include <set>
using namespace std;
static const int MAXN=100;
string s[MAXN+10];
int level[MAXN+10],n;
set<int> ans;
map<string,int> mp;
void dfs(vector<string> vs,int k,int pre,int start)
{
    if(k==vs.size())
    {
        ans.insert(start-1);
        return;
    }
    for(int i=start;level[i]>pre && i<=n;i++)
    {
        if(vs[k][0]=='#' && mp[vs[k]]==i)
        {
            dfs(vs,k+1,level[mp[vs[k]]],mp[vs[k]]+1);
            break;
        }
        if(vs[k]==s[i])
            dfs(vs,k+1,level[i],i+1);
    }
}
int main()
{
    int m;
    cin>>n>>m;
    getchar();
    for(int i=1;i<=n;i++)
    {
        string tmp;
        getline(cin,tmp);
        int pos=0,cnt=0;
        while(tmp[pos]=='.')
        {
            pos++;
            cnt++;
        }
        level[i]=cnt;
        while(tmp[pos]!=' ' && tmp[pos]!='\n' && tmp[pos]!='\0')
            s[i]+=tolower(tmp[pos++]);
        string id="";
        if(tmp[pos]!='\n' && tmp[pos]!='\0') pos++;
        while(tmp[pos]!='\n' && tmp[pos]!='\0')
            id+=tmp[pos++];
        if(id!="")
            mp[id]=i;
    }
    for(int i=1;i<=m;i++)
    {
        string tmp;
        getline(cin,tmp);
        vector<string> find;
        int pos=0;
        string ss="";
        while(pos<tmp.size())
        {
            if(tmp[pos]!=' ' && tmp[pos]!='\n' && tmp[pos]!='\0')
                ss+=tmp[pos];
            else
            {
                find.push_back(ss);
                ss="";
            }
            pos++;
        }
        find.push_back(ss);
        for(int j=0;j<find.size();j++)
            if(find[j][0]=='#')
                continue;
            else
                for(int k=0;k<find[j].size();k++)
                    find[j][k]=tolower(find[j][k]);
        ans.clear();
        dfs(find,0,-1,1);
        cout<<ans.size();
        for(set<int>::iterator it=ans.begin();it!=ans.end();it++)
            cout<<" "<<*it;
        cout<<endl;
    }
    return 0;
}