fork download
  1. #include<bits/stdc++.h>
  2. #define int long long
  3. using namespace std;
  4. int t;
  5. signed main()
  6. {
  7. ios_base::sync_with_stdio(0);
  8. cin.tie(0);cout.tie(0);
  9. cin>>t;
  10. while (t--)
  11. {
  12. map<char,int>b;
  13. string s;int k;
  14. cin>>s>>k;
  15. int dem=0,ans=0;
  16. for (int l=0,r=0;r<s.size();r++)
  17. {
  18. if (b[s[r]]==0)
  19. {
  20. b[s[r]]=1;
  21. }
  22. else b[s[r]]++;
  23. while (b[s[r]] > k)
  24. {
  25. b[s[l]]--;
  26. if (b[s[r]]==k)
  27. {
  28. dem--;
  29. }
  30. l++;
  31. }
  32. ans+=r-l+1;
  33. }
  34. cout<<ans<<"\n";
  35. }
  36. }
  37.  
Success #stdin #stdout 0.01s 5304KB
stdin
1
mfatvfuffzafwaoz
5
stdout
136