Submission #26932

#	Time	Username	Problem	Language	Result	Execution time	Memory
26932		gs14004	CSS (COI14_css)	C++14	20 / 100	5000 ms	26548 KiB

css

#include <bits/stdc++.h>
using namespace std;
typedef long long lint;
typedef pair<int, int> pi;

char buf[5005];

vector<string> readline_split(){
	fgets(buf, 5005, stdin);
	int n = strlen(buf);
	while(n > 0 && buf[n-1] == '\n') n--;
	buf[n] = 0;
	vector<string> ret;
	string cur;
	for(int i=0; i<n; i++){
		if(buf[i] == ' '){
			if(!cur.empty()) ret.push_back(cur);
			cur.clear();
		}
		else cur.push_back(buf[i]);
	}
	if(!cur.empty()) ret.push_back(cur);
	return ret;
}

struct docline{
	int is_end;
	string name;
	vector<string> classes;
};

int n;

docline read_docline(){
	n--;
	auto l = readline_split();
	docline ret;
	if(l.size() == 1 && l[0] == "</div>"){
		ret.is_end = 1;
		return ret;
	}
	ret.is_end = 0;
	ret.name = l[1].substr(4, l[1].size() - 5);
	if(l.size() == 3){
		ret.classes.push_back(l[2].substr(7, l[2].size() - 9));
	}
	else{
		ret.classes.push_back(l[2].substr(7, l[2].size() - 7));
		for(int i=3; i+1<l.size(); i++) ret.classes.push_back(l[i]);
		ret.classes.push_back(l.back().substr(0, l.back().size() - 2));
	}
	return ret;
}

map<string, vector<int> > class_node;
string num_to_name[5005];
int tree_nodes, par[5005];

void make_tree(){
	stack<int> stk;
	stk.push(0);
	while(n && !stk.empty()){
		int x = stk.top();
		docline in = read_docline();
		if(in.is_end == 1) stk.pop();
		else{
			tree_nodes++;
			for(auto &i : in.classes){
				class_node[i].push_back(tree_nodes);
			}
			par[tree_nodes] = x;
			num_to_name[tree_nodes] = in.name;
			stk.push(tree_nodes);
		}
	}
}

int cnt[5005] = {};
vector<int> parse_brj(string s){
	memset(cnt, 0, sizeof(cnt));
	int ok = 0;
	for(int i=0; i<s.size(); ){
		int nxt = i+1;
		while(nxt < s.size() && s[nxt] != '.') nxt++;
		auto x = class_node.find(s.substr(i + 1, nxt - i - 1));
		ok++;
		for(auto &j : x->second){
			cnt[j]++;
		}
		i = nxt;
	}
	vector<int> v;
	for(int i=1; i<=tree_nodes; i++){
		if(cnt[i] == ok) v.push_back(i);
	}
	return v;
}

vector<int> query(vector<string> &s){
	if(s.size() == 1) return parse_brj(s[0]);
	else if(s[s.size() - 2] == ">"){
		auto r = parse_brj(s.back());
		s.pop_back(); s.pop_back();
		auto l = query(s);
		vector<int> ans;
		for(auto &i : r){
			if(binary_search(l.begin(), l.end(), par[i])) ans.push_back(i);
		}
		return ans;
	}
	else{
		auto r = parse_brj(s.back());
		s.pop_back();
		auto l = query(s);
		vector<int> ans;
		for(auto &i : r){
			for(int j=par[i]; j; j=par[j]){
				if(binary_search(l.begin(), l.end(), j)){
					ans.push_back(i);
					break;
				}
			}
		}
		return ans;
	}
}

int main(){
	scanf("%d\n",&n);
	make_tree();
	int m;
	scanf("%d\n", &m);
	while(m--){
		auto in = readline_split();
		auto ret = query(in);
		printf("%d ", ret.size());
		for(auto &i : ret){
			printf("%s ", num_to_name[i].c_str());
		}
		puts("");
	}
}

Compilation message (stderr)

css.cpp: In function 'docline read_docline()':
css.cpp:49:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int i=3; i+1<l.size(); i++) ret.classes.push_back(l[i]);
                   ^
css.cpp: In function 'std::vector<int> parse_brj(std::__cxx11::string)':
css.cpp:82:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0; i<s.size(); ){
                ^
css.cpp:84:13: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   while(nxt < s.size() && s[nxt] != '.') nxt++;
             ^
css.cpp: In function 'int main()':
css.cpp:136:27: warning: format '%d' expects argument of type 'int', but argument 2 has type 'std::vector<int>::size_type {aka long unsigned int}' [-Wformat=]
   printf("%d ", ret.size());
                           ^
css.cpp: In function 'std::vector<std::__cxx11::basic_string<char> > readline_split()':
css.cpp:9:25: warning: ignoring return value of 'char* fgets(char*, int, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
  fgets(buf, 5005, stdin);
                         ^
css.cpp: In function 'int main()':
css.cpp:129:18: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d\n",&n);
                  ^
css.cpp:132:19: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d\n", &m);
                   ^

• Subtask 1: 0 / 10
• Subtask 2: 10 / 10
• Subtask 3: 0 / 10
• Subtask 4: 0 / 10
• Subtask 5: 0 / 10
• Subtask 6: 10 / 10
• Subtask 7: 0 / 10
• Subtask 8: 0 / 10
• Subtask 9: 0 / 10
• Subtask 10: 0 / 10