오늘의 취준/오늘의 코테
[JAVA] 알고리즘 문제풀이 입문 3-4번
gogoem
2023. 5. 31. 14:46
728x90
자바(Java) 알고리즘 문제풀이 입문: 코딩테스트 대비 - 인프런 | 강의
자바(Java)로 코딩테스트를 준비하시는 분을 위한 강좌입니다. 코딩테스트에서 가장 많이 출제되는 Top 10 Topic을 다루고 있습니다. 주제와 연동하여 기초문제부터 중급문제까지 단계적으로 구성
www.inflearn.com
틀린 코드
import java.util.Scanner;
public class Main {
public int solution(int n, int m, int[] arr){
int sum = 0;
int cnt = 0;
int p1 = 0;
int p2 = 0;
while(true){
if(p1 == n-1 && sum < m){
break;
} else {
if(sum < m){
sum += arr[p1++];
if(sum == m){
cnt++;
sum -= arr[p2++];
}
}else if(sum > m){
sum -= arr[p2++];
if(sum == m){
cnt++;
sum -= arr[p2++];
}
}
}
}
return cnt;
}
public static void main(String[] args){
Main t = new Main();
Scanner in = new Scanner(System.in);
int n = in.nextInt();
int m = in.nextInt();
int[] arr = new int[n];
for(int i = 0; i < n; i++){
arr[i] = in.nextInt();
}
in.close();
System.out.print(t.solution(n, m, arr));
}
}
while 문만 사용해서 짜 보려고 했는데 어떤 케이스는 틀리고 어떤 케이스는 정답으로 나옴.
틀리는 케이스는 항상 정답보다 1씩 모자라서 마지막 카운트가 안 되고 있는건가 하고 예상했는데
아무리 봐도 뭐가 문제인지 모르겠어서 일단 강사님께 코드 리뷰 부탁드림..
뭐가 문제일까...
아래는 정답코드
import java.util.Scanner;
public class Main {
public int solution(int n, int m, int[] arr){
int sum = 0;
int cnt = 0;
int p = 0;
for(int i = 0; i < n; i++){
sum += arr[i];
if(sum == m){
cnt++;
sum -= arr[p++];
}
while(sum > m){
sum -= arr[p++];
if(sum == m){
cnt++;
sum -= arr[p++];
}
}
}
return cnt;
}
public static void main(String[] args){
Main t = new Main();
Scanner in = new Scanner(System.in);
int n = in.nextInt();
int m = in.nextInt();
int[] arr = new int[n];
for(int i = 0; i < n; i++){
arr[i] = in.nextInt();
}
in.close();
System.out.print(t.solution(n, m, arr));
}
}