fork download
  1. #include <iostream>
  2.  
  3. using namespace std;
  4. bool check[26] {};
  5. char input[200005];
  6.  
  7. int main()
  8. {
  9. char in;
  10. int n, k,count = 0,sum = 0; cin >> n >> k;
  11. for(int i = 0 ; i < n; ++i)
  12. cin >> input[i];
  13. for (int i = 0; i < k;++i)
  14. {
  15. cin >> in;
  16. check[in - 'a'] = 1;
  17. }
  18.  
  19. for(int i = 0;i < n ;++i)
  20. {
  21. if(check[input[i] - 'a'])
  22. {
  23. ++count;
  24. if(n == (i + 1))
  25. sum += ((count * (count + 1)) / 2);
  26. }
  27. else
  28. {
  29. sum += ((count * (count + 1)) / 2);
  30. count = 0;
  31. }
  32.  
  33. }
  34. cout << sum;
  35.  
  36. }
Success #stdin #stdout 0s 5308KB
stdin
Standard input is empty
stdout
Standard output is empty