인프런 커뮤니티 질문&답변
작성한 질문수
3.2 공통원소 구하기 에서 질문입니다.
작성
·
70
0
샘플문제는 정답인데, 채점사이트에서는 오답으로 나오네요.
어디가 잘못되었는지 못찾겠습니다.
import java.util.*;
class Main {
public ArrayList<Integer> solution(int n, int[] arr1, int m, int[] arr2) {
ArrayList<Integer> answer = new ArrayList<>();
Arrays.sort(arr1);
Arrays.sort(arr2);
int p1=0, p2=0;
while(p1<n && p2<m) {
if(arr1[p1]==arr2[p2]) {
answer.add(arr1[p1]++);
p2++;
}
else if(arr1[p1]<arr2[p2]) p1++;
else p2++;
}
return answer;
}
답변 1
0
안녕하세요, 인프런 AI 인턴입니다.
sulgiro.kwak님, 귀하의 질문에 대한 답변은 오답이 나오는 주된 원인은 반복문 내에서 arr1[p1]의 값을 증가시키는 방식에 문제가 있어 보입니다. 문제의 요구 사항을 정확히 충족시키기 위해 arr1[p1]의 값에 1을 추가하는 대신 p1의 값을 증가시켜야 합니다. 현재 코드에서는 answer.add(arr1[p1]++);라는 라인 때문에 arr1[p1]의 값을 ArrayList에 추가한 후 p1의 값을 1 증체하는 대신 arr1[p1]의 값 자체를 증가시키고 있습니다. 이로 인해 예상치 못한 결과를 초래할 수 있습니다.
따라서, 이 부분을 answer.add(arr1[p1++]);로 수정하면 p1을 증가시킬 수 있습니다. 이렇게 하면 p1과 p2를 올바르게 증가시키면서 공통 원소를 찾을 수 있습니다.
올바른 코드는 다음과 같습니다.
while(p1 < n && p2 < m) {
if(arr1[p1] == arr2[p2]) {
answer.add(arr1[p1]);
p1++;
p2++;
}
else if(arr1[p1] < arr2[p2]) p1++;
else p2++;
}
이렇게 수정하면 공통 원소를 올바르게 찾아내어 정답으로 처리될 가능성이 높습니다.
덕분에 해결했습니다.^^
이게 눈에 안보였네요. ㅎㅎ