fork download
copy 
  1. #include <bits/stdc++.h>
  2.  
  3. using namespace std;
  4. #define int long long
  5. #define ii pair<int,int>
  6. #define f first
  7. #define s second
  8. const int INF=(int)1<<40;
  9. map<int,int > save;
  10. map<int,int> roi_rac_hoa;
  11. signed main()
  12. {
  13.     ios_base::sync_with_stdio(false);
  14.     cin.tie(0);
  15.     int n,k;
  16.     cin>>n>>k;
  17.     int a[n+1];
  18.     for (int i=1;i<=n;i++)
  19.         cin>>a[i];
  20.     int i=1,j=1,dem=1,ans=0;
  21.     save[a[1]]=1;
  22.     while(i<=n)
  23.     {
  24.         if (dem>k||j>n)
  25.         {
  26.             ans+=(j-i);
  27.             save[a[i]]--;
  28.             if (save[a[i]]==0)
  29.                 dem--;
  30.             i++;
  31.         }
  32.         else {
  33.             j++;
  34.             if (j<=n)
  35.             {
  36.                 save[a[j]]++;
  37.                 if (save[a[j]]==1)
  38.                     dem++;
  39.             }
  40.         }
  41.     }
  42.     cout<<ans;
  43.     return 0;
  44. }
Success #stdin #stdout 0s 5280KB
comments (?)
stdin
copy 
Standard input is empty
stdout
copy 
1
https://ideone.com/UKhfBX
language:
C++ (gcc 8.3)
created:
10 days 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!