fork download
copy 
  1.     #include<bits/stdc++.h>
  2.     using namespace std;
  3.     typedef long long int ll;
  4.    const ll MO = 1e9+7;
  5.  
  6.     int main() {
  7.             ios_base::sync_with_stdio(false);
  8.             cin.tie(NULL);
  9.             cout.tie(NULL);
  10.  
  11.             ll t;
  12.             cin>>t;
  13.             while(t--)
  14.             {
  15.             ll n,k;
  16.             cin>>n>>k;
  17.             string str;
  18.             cin>>str;
  19.             vector<ll> v1,v2,v3;
  20.             ll rd=0;
  21.             for(ll i=0;i<str.size();++i)
  22.             {
  23.                 if(str[i]=='b')
  24.                     rd++;
  25.             }
  26.             //cout<<rd<<endl;
  27.             ll co=0;
  28.             for(ll i=0;i<str.size();++i)
  29.             {
  30.                 if(str[i]=='a')
  31.                     {
  32.                         //v1.push_back(i+1);
  33.                        v3.push_back(rd-co);
  34.                     }
  35.                 else if(str[i]=='b')
  36.                     {
  37.                         co++;
  38.                     }
  39.             }
  40.             unsigned long long int dt=0;
  41.             unsigned long long int gt=k*(k-1)/2;
  42.             for(ll i=0;i<v3.size();++i)
  43.             {
  44.                 dt+=(k*v3[i]+gt*rd);
  45.             }
  46.             //unsigned long long int gt=k*(k+1)/2;
  47. //dt*=gt;
  48.             cout<<dt<<endl;
  49.     }
  50.     return 0;
  51.     }
  52.  
Success #stdin #stdout 0s 15240KB
comments (?)
stdin
copy 
3
4 2
abcb
7 1
aayzbaa
12 80123123
abzbabzbazab
stdout
copy 
6
2
64197148392731290
https://ideone.com/Div8Ty
language:
C++14 (gcc 8.3)
created:
6 years ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!