프로그래머스 - java 리스트 자르기

💡 좋은 코드

 

초반에는 Arrays.copyofRange 함수를 통해서 간단하게 리스트를 잘라줬고

이후 n == 4 일 때에는 동적배열을 불러오고 차례대로 삽입해준다음에

동적배열을 정수배열로 변환해주면 끝 !!

import java.util.*;
class Solution {
    public int[] solution(int n, int[] slicer, int[] num_list) {
        int a = slicer[0];
        int b = slicer[1];
        int c = slicer[2];
        if(n ==1)
            return Arrays.copyOfRange(num_list, 0, b+1);
        if(n==2)
            return Arrays.copyOfRange(num_list, a, num_list.length);
        if(n==3)
            return Arrays.copyOfRange(num_list, a, b+1);
            
      List<Integer> result = new ArrayList<>();  // n = 4일때 따로 담아줄 동적 배열
      
        for (int i = a; i <= b; i += c) { 
            result.add(num_list[i]); // 동적 배열에 추가
        }
        int[] answer = new int[result.size()];  // 정수배열로 반환해야하므로 변환시키기
        for (int i = 0; i < result.size(); i++) { 
            answer[i] = result.get(i); // 삽입해주면 끝
        }
        return answer;
    }
}

 

1. Arrays.copyOf() :  배열 전체를 복사
2. Arrays.copyOfRange() :  특정 범위를 복사합니다. 
3. IntStream.range() :  정수 스트림을 생성하는 메서드
4. map()  각 요소에 대해 주어진 함수를 적용
5. toArray()는 스트림의 요소를 배열로 변환하는 최종 연산

import java.util.Arrays;
import java.util.stream.IntStream;

class Solution {
    public int[] solution(int n, int[] slicer, int[] num_list) {

        int[] answer = null;
        int a = slicer[0], b = slicer[1], c = slicer[2]; 

        switch (n) {
        case 1:
            answer = Arrays.copyOf(num_list, b + 1); // 처음 ~ b 
            break;
        case 2:
            answer = Arrays.copyOfRange(num_list, a, num_list.length); // a ~ 끝까지
            break;
        case 3:
            answer = Arrays.copyOfRange(num_list, a, b + 1); //a, b 까지
            break;
        case 4:
            answer = IntStream.range(0, (b - a) / c + 1).map(i -> num_list[a + i * c]).toArray();
            // 0부터 (b-a)/c+1 까지 정수 스트림 생성
            // 앞에서 선택된 요소 i => 를  a+i*c 로 계산 후 => 선택된 값으로 스트림 생성 후
            // 배열로 반환
            break;
        }

        return answer;
    }
}

 

 

---

 

😭 1차 오류 : switch 문에서 case 마다 break문을 빠뜨렸다.

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

class Solution {
    public  ArrayList <Integer> solution(int n, int[] slicer, int[] num_list) {
            ArrayList <String> str_list = new ArrayList<String>();
            ArrayList <Integer> answer = new ArrayList<>();
        
            String str = Arrays.toString(num_list); //배열 문자열로 바꾸기
            str = str.replace("[", "").replace("]", "").replace(",","").replace(" ","");
        
            int a = slicer[0];
            int b = slicer[1];
            int c = slicer[2];
            
        switch (n) {
            case 1:
                str_list.add(str.substring(0,b+1));
            case 2:
                str_list.add(str.substring(a,num_list.length));
            case 3:
                str_list.add(str.substring(a,b+1));
            case 4:
                for (int i = a; i <= b; i+=c){ //1~5  
                    str_list.add(String.valueOf(num_list[i]));
                }              
            }
            
            for (int j = 0; j<str_list.size(); j++){
                answer.add(Integer.parseInt(str_list.get(j)));
            }
        
        return answer;
    }
}

 

 

 

😭2차 오류 : 이런저런 문자열 <-> 정수 변환을 해결했는데 ...!!!

테스트 케이스는 맞았지만 제출하면 오류가 생긴다 ㅠ

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

class Solution {
    public  ArrayList <Integer> solution(int n, int[] slicer, int[] num_list) {
            ArrayList <String> str_list = new ArrayList<String>();
            ArrayList <Integer> answer = new ArrayList<>();
            String tmp = "";
        
            String str = Arrays.toString(num_list); //배열 문자열로 바꾸기
            str = str.replace("[", "").replace("]", "").replace(",","").replace(" ","");
        
            int a = slicer[0];
            int b = slicer[1];
            int c = slicer[2];
            
        switch (n) {
            case 1:
                tmp = str.substring(0,b+1); // 문자열에 삽입
                   break;
            case 2:
                tmp = str.substring(a,num_list.length);
                   break;
            case 3:
                tmp = str.substring(a,b+1);
                   break;
            case 4:
                for (int i = a; i <= b; i+=c){ //1~5  
                    str_list.add(String.valueOf(num_list[i]));
                }        
                   break;
            }
            
        
                if ( n == 1 || n == 2 || n == 3){
                for (int j = 0; j < tmp.length(); j++){
                    char ch = tmp.charAt(j); //문자 하나씩 가져오기
                    answer.add(Integer.valueOf(""+ch)); //
                }
                    
                }else {
            for (int j = 0; j < str_list.size(); j++){
              answer.add(Integer.parseInt(str_list.get(j)));
            }
        }
               
        
        return answer;
    }
}