Algorithm/문제

34. 모든 아나그램 찾기

챛채 2023. 9. 5. 17:44

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Scanner;
import java.util.Map;

public class Main {

    public int solution(String a, String b) {
        int answer = 0;

        HashMap<Character, Integer> am = new HashMap<>();
        HashMap<Character, Integer> bm = new HashMap<>();

        //미리 셋팅 
        for(char x: b.toCharArray()) bm.put(x, bm.getOrDefault(x, 0) + 1);

        int L = b.length() - 1;
        for(int i = 0; i < L; i++) am.put(a.charAt(i), am.getOrDefault(a.charAt(i), 0) + 1);
        int lt =0;
        for(int rt = L; rt < a.length(); rt++) {
            am.put(a.charAt(rt), am.getOrDefault(a.charAt(rt), 0)  + 1);
            if(am.equals(bm)) answer++;
            am.put(a.charAt(lt), am.get(a.charAt(lt)) - 1);
            if(am.get(a.charAt(lt)) == 0) am.remove(a.charAt(lt));
            lt++;
        }
       return answer;
    }
   public static void main(String[] args){
       Main T = new Main();
       Scanner kb = new Scanner(System.in);
       String a = kb.next();
       String b = kb.next();
       System.out.println(T.solution(a, b));
   }
}

 

배열 인덱스를 k-1개를 미리 세팅해 놓는 이유?

-> 미리 k-1개를 해싱해두고 그 다음 for문 부터 rt로 하나 추가해서 k개를 해싱해 아나그램인지 확인하고, lt로 하나 빼주고 하면서 계속 밀고 나가기 위함