月模拟题 CSP 201809-3 元素选择器

题目描述

样例

Input

Output

样例输入

样例输出

思路

每一行都采用结构体来存储，分别记录标签、id和层数
对于文档的读入，使用getline读入一行，先对 . 的个数进行计算，更新层数，然后对label进行判断，由于label大小写不敏感，可以全部换成小写，最后对id进行判断
对于查询的读入，将读入的一行信息通过判断空格进行分隔并依次放入vector中，将所有不以#开头的标签全部转换为小写，对于查询，从后向前遍历文档，如果找到符合要求的元素，继续向前判断其祖先是否符合，如果符合，将元素的层数放入ans中
输出最后结果即可

代码

#include <iostream>
#include <string>
#include <vector>
using namespace std;

int n,m;

struct Node
{
	string id;//id属性 
	string label;//标签 
	int num;//层数 
	Node(){ }
}node[110];

vector<string> v;//用来存储查询的语句
vector<int> ans;//存储每个查询的答案 即位于第几行

bool charge(int &index,int &num,int size)
{
	for(int i=index;i>=1;i--)
	{
		if(node[i].num<num)
		{
			num=node[i].num;
			index=i;
			if(node[i].id==v[size] || node[i].label==v[size])
				return true;
		}
	}
	return false;
}
 
int main()
{
	cin>>n>>m;
	getchar();
	for(int i=1;i<=n;i++)
	{
		string str;
		getline(cin,str);
		int sum=0;//记录.的个数 
		while(str[sum]=='.')
			sum++;
		node[i].num=sum/2;
		
		string label;
		
		while(str[sum]!=' ' && sum<str.size())//找到label 
			label=label+str[sum], sum++;
			
		for(int j=0;j<label.size();j++)//label不区分大小写，全部换成小写
		{
			if(label[j]>='A'&&label[j]<='Z')
			{
				label[j]=label[j]-'A'+'a';
			}
		} 
		node[i].label=label;
		
		if(sum<str.size())//此时有id
		{
			string id;
			sum++;
			while(sum<str.size())
				id=id+str[sum],sum++;
			node[i].id=id;		 
		} 
	}
	
	//查找
	for(int i=1;i<=m;i++)
	{
		string str;
		getline(cin,str);
		v.clear();
		ans.clear();
		
		int sum=0;
		while(sum<str.size())//将读入的查询指令进行分开 
		{
			string s;
			while(sum<str.size()&&str[sum]!=' ')
				s=s+str[sum],sum++;
			v.push_back(s);
			if(sum<str.size()&&str[sum]==' ')
				sum++;
		}	
		
		for(int j=0;j<v.size();j++)//将所有的label变小写 
		{
			if(v[j][0]!='#') //id前面有# 无需变大小写 
			{
				for(int k=0;k<v[j].size();k++)
				{
					if(v[j][k]>='A' && v[j][k] <='Z')
						v[j][k]=v[j][k]-'A'+'a';
				}
			}
		}	
	
		//从后向前遍历，找到符合标准的元素，判断其祖先是否符合标准
		for(int j=n;j>=1;j--)
		{
			if(node[j].id==v[v.size()-1] || node[j].label==v[v.size()-1])
			{
				int index=j;
				int numm=node[j].num;
				int sizee=v.size()-2;
				while(sizee>=0)
				{
					if(!charge(index,numm,sizee)) break;
					sizee--;
				}
				if(sizee==-1) ans.push_back(j);
			}
		} 
		
		cout<<ans.size();
		for(int j=ans.size()-1;j>=0;j--)
		{
			cout<<" "<<ans[j];
		}
		cout<<endl;		
	} 
}