인프런 커뮤니티 질문&답변

이주용님의 프로필 이미지
이주용

작성한 질문수

자바(Java) 알고리즘 문제풀이 입문: 코딩테스트 대비

3. 매출액의 종류(Hash, sliding window)

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

작성

·

286

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

이주용님의 프로필 이미지
이주용
질문자

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

이주용님의 프로필 이미지
이주용

작성한 질문수

질문하기