fork download
copy 
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. #define ll long long
  4.  
  5. int main() {
  6. 	ll n;
  7.     cin>>n;
  8.     ll k;cin>>k;
  9.     ll b[n];
  10.     for(ll i=0;i<n;i++){
  11.         cin>>b[i];
  12.     }ll count = 0 ;
  13.  
  14.  
  15.     ll sum = 0 ;
  16.     unordered_map <ll,ll> g ; 
  17.     for (int i = 0, j = 0; j < n; j++) {
  18.             g[b[j]]=g[b[j]]+1;
  19.             ll d = g.size();
  20.             while (d>k){
  21.                 g[b[i]] = g[b[i]] - 1;
  22.                 if(g[b[i]]==0){
  23.                     g.erase(b[i]);
  24.                 }
  25.                 i++;
  26.  
  27.                 d = g.size();
  28.  
  29.             }
  30.             count += (j - i + 1);
  31.      }
  32.  
  33.     cout<<count;
  34.     return 0;
  35. }
Success #stdin #stdout 0.01s 5280KB
comments (?)
stdin
copy 
Standard input is empty
stdout
copy 
2642899456
https://ideone.com/1VEVZX
language:
C++14 (gcc 8.3)
created:
1 day 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!