[프로그래머스] 글자 지우기 (Java, Python)

문제

https://school.programmers.co.kr/learn/courses/30/lessons/181900

프로그래머스

 

코드(Java)

import java.util.HashSet;
import java.util.Set;

class Solution {
    public String solution(String my_string, int[] indices) {
        Set<Integer> indexSet = new HashSet<>();
        for (int index : indices) {
            indexSet.add(index);
        }
        
        StringBuilder sb = new StringBuilder();
        
        for (int i = 0; i < my_string.length(); i++) {
            if(!indexSet.contains(i)) {
                sb.append(my_string.charAt(i));
            }
        }
        
        return sb.toString();
    }
}

 

코드(Python)

def solution(my_string, indices):
    index_set = set(indices)
    return ''.join(my_string[i] for i in range(len(my_string)) if i not in index_set)