Rhyming Slang

#include <iostream>
#include <string>
#include <vector>
#include <set>

using namespace std;

vector<string> split(string s) {
	vector<string> v;
	while (!s.empty()) {
		auto p = s.find(' ');
		if (p == string::npos) {
			v.push_back(s);
			break;
		}
		v.push_back(s.substr(0, p));
		s = s.substr(p+1);
	}
	return v;
}

bool isSuffix(string a, string b) {
	if (a.size() < b.size())
		return false;
	return a.substr(a.size()-b.size()) == b;
}

int main() {
	string tmp;
	string s;
	getline(cin, s);
	int e;
	cin >> e;
	getline(cin, tmp);
	set<string> rhymes;
	for (int i = 0; i < e; i++) {
		string l;
		getline(cin, l);
		auto ps = split(l);
		for (string p : ps) {
			if (isSuffix(s, p)) {
				rhymes.insert(ps.begin(), ps.end());
				break;
			}
		}
	}
	
	int n;
	cin >> n;
	getline(cin, tmp);
	
	for (int i = 0; i < n; i++) {
		string l;
		getline(cin, l);
		bool found = false;
		for (string r : rhymes)
			if (isSuffix(l, r))
				found = true;
		if (found)
			cout << "YES\n";
		else
			cout << "NO\n";
	}
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED