• 카테고리

    질문 & 답변
  • 세부 분야

    알고리즘 · 자료구조

  • 해결 여부

    미해결

set을 이용하여 풀었는데 시간초과가 뜹니다.

23.12.16 21:20 작성 조회수 196

0

package hash;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.Scanner;

public class TypeOfSales {

	static ArrayList<Integer> solution(int n, int k, int[] arr) {
		
		Collection<Integer> set = new HashSet<>();
		Collection<Integer> list = new ArrayList<>();
		ArrayList<Integer> result = new ArrayList<>();		
		int p1 = 1;
		
		for(int i = 0; i < k; i++) {
			
			list.add(arr[i]);
			
		}
		set.addAll(list);
		result.add(set.size());
		
		while(p1 < n-k+1) {
			
			set.clear();
			
			list.remove(arr[p1-1]);
			list.add(arr[p1+k-1]);
			
			set.addAll(list);
			result.add(set.size());
			
			p1++;
			
		}
		
		return result;
		
	}
	
	public static void main(String[] args) {
		
		Scanner sc = new Scanner(System.in);
		
		int n = sc.nextInt();
		int k = sc.nextInt();
		int[] arr = new int[n];
		
		for(int i = 0; i < n; i++) {
			
			arr[i] = sc.nextInt();
			
		}
		for(int x : TypeOfSales.solution(n, k, arr)) {
			System.out.print(x + " ");
		}
	}

}

시간 복잡도는 O(N)이 맞는거같은데 4번 5번 테스트 케이스에서 2초 가까이 뜨네용..

답변 1

답변을 작성해보세요.

0

이주용님의 프로필

이주용

질문자

2023.12.16

혹시몰라서 LinkedList로 바꾸어서 했는데도 별차이 없군요.. 그냥 강의대로 푸는게 최선인거같습니다...