fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main()
  5. {
  6. string s;
  7. cin >> s;
  8.  
  9. set<int> possible_val;
  10.  
  11. int c = 1;
  12. for (size_t i = 1; i < s.length(); i++) {
  13. if (s[i] == s[i - 1]) {
  14. c++;
  15. }
  16. else {
  17. for (int j = 1; j <= c; j++)
  18. possible_val.insert(j * (s[i - 1] - 'a' + 1));
  19. c = 1;
  20. }
  21. }
  22. for (int j = 1; j <= c; j++)
  23. possible_val.insert(j * (s[s.length() - 1] - 'a' + 1));
  24. //for (int x : possible_val) cout << x << endl;
  25.  
  26. int q, x;
  27. cin >> q;
  28. while (q-- > 0) {
  29. cin >> x;
  30. if (possible_val.find(x) != possible_val.end()) cout << "YES\n";
  31. else cout << "NO\n";
  32. }
  33. return 0;
  34. }
Success #stdin #stdout 0s 15240KB
stdin
abccddde
6
1
3
12
5
9
10
stdout
YES
YES
YES
YES
NO
NO