TIL (2024.06.16)

문제 1️⃣ (🔗 리트 코드 _ 특정 문자를 포함한 문자열 찾기)

💡 IDEA : String 내부 함수인 indexOf 매서드르 통해 인덱스를 확인하고, -1 인 경우 해당 문자를 포함하지 않는 것으로 판단하여 나머지만 인덱스값을 정답 배열에 추가한다.

import java.util.List;
import java.util.ArrayList;

class Solution {
    public List<Integer> findWordsContaining(String[] words, char x) {
        List<Integer> answer = new ArrayList<>();

        for (int i=0; i<words.length; i++) {
            if (words[i].lastIndexOf(x) >= 0) {
                answer.add(i);
            }
        }

        return answer;
    }
}

문제 2️⃣ (🔗 리트 코드 _ 사이즈 그룹에 맞게 팀원 구성하기)

💡 IDEA : 희망 그룹 명 수를 확인할 수 있는 List<Integer>[] 배열을 선언해서 해당 인덱스 값들을 알맞게 넣은 후 리스트로 부터 차례대로 희망 그룹 수 만큼 구성하여 answer 리스트에 추가한다.

import java.util.ArrayList;
import java.util.List;
import java.util.Arrays;

class Solution {
    public List<List<Integer>> groupThePeople(int[] groupSizes) {
        int[] _groupSizes = Arrays.copyOf(groupSizes, groupSizes.length);
        Arrays.sort(_groupSizes);

        // (1) 각 그룹에 해당하는 index 취합을 위한 배열 선언
        int min = _groupSizes[0];
        int max = _groupSizes[_groupSizes.length-1];

        List<Integer>[] count = new ArrayList[max+1];

        for (int i=0; i<count.length; i++) {
            count[i] = new ArrayList<>();
        }

        // (2) index 취합
        for (int j=0; j<groupSizes.length; j++) {
            count[groupSizes[j]].add(j);
        }

        // (3) 조합 후 answer 리스트에 담기
        List<List<Integer>> answer = new ArrayList<>();

        int size = 0;
        ArrayList<Integer> arr = new ArrayList<>();

        for (int i=min; i<count.length; i++) {
            List<Integer> cnt = count[i];
            size = cnt.size();

            if (size == 0) continue;

            for (int j=0; j<size; j++) {
                System.out.println(j%i);
                if (j % i == 0) {
                    arr = new ArrayList<>();
                }

                arr.add(cnt.get(j));
                System.out.println(cnt.get(j));

                if ((j+1) % i == 0) {
                    answer.add(arr);
                }
            }
        }

        return answer;
    }
}

➡️ 통과하기는 했지만 런타임이 평균에 비해 긴 편이라 좀 더 짧은 로직을 생각해보아야 할 것 같다.