[백준] 2003 수들의 합2 투 포인터

문제 :

www.acmicpc.net/problem/2003

2003번: 수들의 합 2

분석 :

투 포인터란 1차원 배열에 대해 포인터를 양 끝단에 두고 값을 찾는 방식을 말한다.

조건(m)에 해당 되는 값을 찾으면 cnt를 증가

m 보다 작은 값이면 end 인덱스 위치의 값을 sum에 저장하고 end 하나 증가

m 보다 큰 값이면 start 인덱스의 값을 빼고 start를 하나 증가시킴 

 

c++ 코드 :

//
//  2003_sum2.cpp
//  SOMA👩🏻‍💻
//
//  Created by JoSoJeong on 2021/02/26.
//

#include <stdio.h>
#include <iostream>
#include <vector>


using namespace std;

int two_pointer(int N, int arr[], int M){
    int cnt = 0;
    int start = 0, end = 0;
    int sum = 0;
    
    while(true){
        
        if(sum >= M){
            sum = sum - arr[start++]; // 값을 빼주고 start 인덱스 뒤로 감
        }else if(end == N){ // end 가 인덱스 범위 벗어났을 경우
            break;
        }else{ // end 현재 위치값 저장 후 end 뒤로 이동
            sum = sum + arr[end++];
        }
        
        if(sum == M){
            cnt++;
        }
    }
    
    return cnt;
}
int main(){
    int N, M;
    cin >> N >> M;
    int arr[N];
    for(int i =0; i < N; i++){
        cin >> arr[i];
    }
  
    int cnt = two_pointer(N, arr, M);
    cout << cnt << endl;
    return 0;
}