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

최수연님의 프로필 이미지
최수연

작성한 질문수

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

4. 모든 아나그램 찾기(Hash, sliding window : 시간복잡도 O(n))

4. 모든 아나그램 찾기 질문

작성

·

204

0

안녕하세요 다음 제 풀이가 테스트케이스 1,2번에서만 오류가 나는데 코드의 어느 부분이 잘못되었는지 모르겠어서 질문 드리고 싶습니다.

import java.util.HashMap;
import java.util.Scanner;

public class Main {
    public static int solution(String S,String T){
        int answer = 0;
        HashMap<Character,Integer> ThashMap = new HashMap<>();
        for (char x: T.toCharArray()){
            ThashMap.put(x, ThashMap.getOrDefault(x, 0)+1);
        }

        HashMap<Character,Integer> ShashMap = new HashMap<>();
        // 1번째 윈도우
        for (int i=0; i<T.length(); i++){
            ShashMap.put(S.charAt(i), ShashMap.getOrDefault(S.charAt(i), 0)+1);
        }
        if (ThashMap.equals(ShashMap)) answer ++;

        // 나머지 윈도우
        for (int i=T.length(); i<S.length()-1; i++){
            ShashMap.put(S.charAt(i), ShashMap.getOrDefault(S.charAt(i), 0)+1);
            ShashMap.put(S.charAt(i-T.length()), ShashMap.get(S.charAt(i-T.length()))-1);
            if (ShashMap.get(S.charAt(i-T.length())) == 0) ShashMap.remove(S.charAt(i-T.length()));
            if (ThashMap.equals(ShashMap)) answer ++;
        }

        return answer;
    }
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String str1 = sc.nextLine();
        String str2 = sc.nextLine();
        System.out.println(solution(str1,str2));
    }
}


답변 1

0

김태원님의 프로필 이미지
김태원
지식공유자

안녕하세요^^

문자열의 끝에 있는 아나그램을 확인하지 못하고 있는 코드였습니다. for문을 아래처럼 바꾸면 됩니다.

for (int i=T.length(); i<S.length(); i++){
최수연님의 프로필 이미지
최수연
질문자

정말 감사합니다!!

최수연님의 프로필 이미지
최수연

작성한 질문수

질문하기