inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

자바(Java) 알고리즘 문제풀이 입문: 코딩테스트 대비

4. 연속부분수열(복합적 문제)

연속분수수열 코드리뷰 부탁드립니다 ㅠ

348

티라노사우루스

작성한 질문수 2

0

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        //입력
        Scanner scn = new Scanner(System.in);
        int n = scn.nextInt();
        int m = scn.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = scn.nextInt();
        }

        int count = 0;
        int sum = 0;
        int i = 0;
        int k = 1; // 더한 수열 인자의 갯수
        while(i<n){
            sum += arr[i];
            if(sum<m) {
                i++; k++;
            } else if(sum>m){
                while(sum>m){
                    sum -= arr[i-(k-1)]; // m 보다 크면 맨앞에거 창문에서 제거
                    k--;
                    if(sum==m) count++; i++; k++; // 맨앞을빼줘서 count증가하고 index증가할때 그 다음이 바로 sum>m 경우면 또 k--를 빼주기때문에 여기서 k++를 해줘야한다.
                    if(sum<m) i++;
                }
            } else if(sum==m){
                count++;
                i++;
            }
        }

        System.out.println(count);
    }
}

문제 예제와 다른 커뮤니티 질문에 답변에 있던 예제

5 3
1 1 1 1 1

경우도 통과했는데 오답이 나옵니다 ㅠㅠ

 

저는 lt, rt 같은 위치자(?)를 사용하지 않고

sum 되고 있는 배열 인덱스의 갯수를 세서

sum > m 경우 맨 앞인덱스를 제외시킬때 사용하였습니다.

java 코딩-테스트

답변 1

2

김태원

안녕하세요^^

반례입니다.

5 2
1 3 1 1 1

답은 2인데 0이 나옵니다.

다른 반례들도 스스로 만들어서 테스트해보세요.

그리고 영상의 방법을 꼭 익혀두세요. 요즘은 영상의 방법을 살짝 개선해서 아래와 같이 짜고 있습니다.

import java.util.*;
class Main {	
	public int solution(int n, int m, int[] arr){
		int answer = 0, sum = 0, lt = 0;
		for(int rt = 0; rt < n; rt++){
			sum += arr[rt];
			while(sum > m){
				sum -= arr[lt++];
			}
			if(sum == m) answer++;
		}
		return answer;
	}

	public static void main(String[] args){
		Main T = new Main();
		Scanner kb = new Scanner(System.in);
		int n=kb.nextInt();
		int m=kb.nextInt();
		int[] arr=new int[n];
		for(int i=0; i<n; i++){
			arr[i]=kb.nextInt();
		}
		System.out.print(T.solution(n, m, arr));
	}
}

 

0

티라노사우루스

감사합니다.!!

 

안녕하세요. 바뀐 채점사이트 관련해서 문의드립니다.

0

26

1

갑자기 채점 사이트가 바뀌었어요

0

31

1

문제 리스트 페이지

0

26

1

채점 사이트 관련 질문드립니다

0

23

1

봉우리 문제 질문입니다

0

80

2

씨름 선수 문제에서 각 선수의 몸무게나 키가 같을 수도 있다면?

0

64

0

이 코드랑 영상 코드중에 뭐가 더 좋은 코드인가요?

0

71

0

가중치 방향 그래프에서 가중치가 0인 간선을 표현하는 방법

0

67

1

좌표 정렬 문제 이 코드가 왜 틀린지 모르겠습니다 ㅠㅠ

0

84

2

6-7 강의에서

0

48

1

6-6. 장난꾸러기 질문 있습니다.

0

45

1

강의 수강후 코딩테스트

0

109

1

answer 변수 사용 여부

0

43

1

2중 for문

1

84

2

2-11. 임시반장정하기 (Runtime Error)

0

63

1

혹시 LinkedList 같은 자료 구조들은 따로 배우지 않나요?

0

70

1

이런 풀이는 어떨까요

0

43

1

자바 스트림 방식의 효율성 질문 드립니다.

0

57

1

알고리즘 자료 구조들..

0

62

1

StringBuilder vs BufferdWriter

0

47

1

원더랜드(프림)

0

49

1

이런 코드는 어떤가요?

0

61

1

bfs 풀이

0

57

1

병합정렬

0

55

1