fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main() {
  5. string str;
  6. int i,q,j,test,x,k;
  7. bool temp;
  8. vector<int> vect;
  9. cin>>str;
  10. for(i=0;i<str.length();i++){
  11. for(j=i+1;j<str.length();j++){
  12. if(str[i] != str[j] ){
  13.  
  14. for(k=1;k<=j-i;k++){
  15. vect.push_back((str[i]-96)*k);
  16. }
  17. i=j-1;
  18. break;
  19. }
  20. if(str[i] == str[j] && j==str.length()-1){
  21. // printf("Hey %d %d\n",i,j);
  22. for(k=1;k<=j-i+1;k++){
  23. vect.push_back((str[i]-96)*k);
  24. // printf("R %d\n",(str[i]-96)*k);
  25. }
  26. i=j-1;
  27. break;
  28. }
  29. }
  30. }
  31. if(str[str.length()-1]!=str[str.length()-2]){
  32. vect.push_back(str[str.length()-1]-96);
  33. }
  34. // for(i=0;i<vect.size();i++){
  35. // printf("%d\n",vect[i]);
  36. // }
  37. scanf("%d",&test);
  38.  
  39. while(test--){
  40. scanf("%d",&x);
  41. temp = false;
  42. for(i=0;i<vect.size();i++){
  43. if(vect[i] == x){
  44. temp = true;
  45. break;
  46. }
  47. }
  48. if(temp == true){
  49. printf("YES\n");
  50. }
  51. else{
  52. printf("NO\n");
  53. }
  54. }
  55. return 0;
  56. }
Success #stdin #stdout 0s 4392KB
stdin
aaabbbbcccddd
5
9
7
8
12
5
stdout
YES
NO
YES
YES
NO