오늘의 취준/오늘의 코테
[JAVA] 알고리즘 문제풀이 입문 3-1번
gogoem
2023. 5. 29. 16:23
728x90
자바(Java) 알고리즘 문제풀이 입문: 코딩테스트 대비 - 인프런 | 강의
자바(Java)로 코딩테스트를 준비하시는 분을 위한 강좌입니다. 코딩테스트에서 가장 많이 출제되는 Top 10 Topic을 다루고 있습니다. 주제와 연동하여 기초문제부터 중급문제까지 단계적으로 구성
www.inflearn.com
import java.util.ArrayList;
import java.util.Scanner;
public class Main {
public ArrayList<Integer> solution(int n, int m, int[] arr1, int[] arr2){
ArrayList<Integer> answer = new ArrayList<>();
int p1 = 0, p2 = 0;
while(p1<n && p2<m){
if(arr1[p1]>arr2[p2]){
answer.add(arr2[p2]);
p2++;
}else{
answer.add(arr1[p1]);
p1++;
}
}
while(p1<n){
answer.add(arr1[p1]);
p1++;
}
while(p2<m){
answer.add(arr2[p2]);
p2++;
}
return answer;
}
public static void main(String[] args){
Main t = new Main();
Scanner in = new Scanner(System.in);
int n = in.nextInt();
int[] arr1 = new int[n];
for(int i = 0; i < n; i++){
arr1[i] = in.nextInt();
}
int m = in.nextInt();
int[] arr2 = new int[m];
for(int i = 0; i < m; i++){
arr2[i] = in.nextInt();
}
in.close();
for(int x : t.solution(n, m, arr1, arr2)){
System.out.print(x+" ");
}
}
}
* 배열 인덱스 범위 항상 신경쓰기
* 전/후위증감연산자 사용하기
* 기본 배열보다 ArrayList 등 클래스 사용하는 습관 들이기
수정코드
import java.util.ArrayList;
import java.util.Scanner;
public class Main {
public ArrayList<Integer> solution(int n, int m, int[] arr1, int[] arr2){
ArrayList<Integer> answer = new ArrayList<>();
int p1 = 0, p2 = 0;
while(p1<n && p2<m){
if(arr1[p1]>arr2[p2]){
answer.add(arr2[p2++]);
}else{
answer.add(arr1[p1++]);
}
}
while(p1<n){ answer.add(arr1[p1++]); }
while(p2<m){ answer.add(arr2[p2++]); }
return answer;
}
public static void main(String[] args){
Main t = new Main();
Scanner in = new Scanner(System.in);
int n = in.nextInt();
int[] arr1 = new int[n];
for(int i = 0; i < n; i++){
arr1[i] = in.nextInt();
}
int m = in.nextInt();
int[] arr2 = new int[m];
for(int i = 0; i < m; i++){
arr2[i] = in.nextInt();
}
in.close();
for(int x : t.solution(n, m, arr1, arr2)){
System.out.print(x+" ");
}
}
}