fork download 
  1. /* package whatever; // don't place package name! */
  2.  
  3. import java.util.*;
  4. import java.lang.*;
  5. import java.io.*;
  6.  
  7. /* Name of the class has to be "Main" only if the class is public. */
  8. class Ideone
  9. {
  10. 	public static void main (String[] args) throws java.lang.Exception
  11. 	{
  12. 		// your code goes here
  13. 		Scanner sc=new Scanner(System.in);
  14. 		int n=sc.nextInt();
  15. 		int arr[]=new int[n];
  16. 		int k=sc.nextInt();
  17. 		for(int i=0;i<n;i++)
  18. 		arr[i]=sc.nextInt();
  19. 		int count=0;
  20. 		Map<Integer,Integer> m=new HashMap<>();
  21. 		for(int i=0,j=0;j<n;j++)
  22. 		{
  23. 			m.put(arr[j],m.getOrDefault(arr[j],0)+1);
  24. 			int size=m.size();
  25. 			while(size>k)
  26. 			{
  27. 				m.put(arr[i],m.get(arr[i])-1);
  28. 				if(m.get(arr[i])==0)
  29. 				m.remove(arr[i]);
  30. 				size=m.size();
  31. 				i++;
  32. 			}
  33. 			count+=(j-i+1);
  34. 		}
  35.  
  36. 		System.out.println(count);
  37.  
  38.  
  39. 	}
  40. }
Success #stdin #stdout 0.11s 54488KB
comments (?)
stdin 
3 3 
1 2 3
stdout 
6
https://ideone.com/Z243Nt
language:
Java (HotSpot 12)
created:
13 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!