fork download
  1.  
  2. #include <bits/stdc++.h>
  3. using namespace std;
  4.  
  5. int main()
  6. {
  7. string g;
  8. int n, k;
  9. cin >> n >> k >> g;
  10.  
  11. int count = 0;
  12.  
  13. for (int i = 0; i <= n - 1; i++)
  14. {
  15. unordered_map<int, int> b;
  16.  
  17. for (int j = i; j <= n - 1; j++)
  18. {
  19. int number = g[j] - '0';
  20. b[number] = b[number] + 1;
  21.  
  22. // Check whether substring i...j is valid
  23. int u = 0;
  24.  
  25. for (int i = 0; i <= 9; i++)
  26. {
  27. if (b[i] == k || b[i] == 0)
  28. {
  29. // Valid frequency
  30. }
  31. else
  32. {
  33. u = 1;
  34. break;
  35. }
  36. }
  37.  
  38. if (u == 0)
  39. {
  40. count = count + 1;
  41. }
  42. }
  43. }
  44.  
  45. cout << count;
  46.  
  47. return 0;
  48. }
  49.  
Success #stdin #stdout 0s 5320KB
stdin
6
2
110022
stdout
6