백준 2003 수들의 합 2 c++

 

 

 

처음 작성한 코드는 다음과 같다.

하지만 틀렸다는 결과가 나왔다.

#include <iostream>
using namespace std;

int main(){
    int n,m;
    cin >> n >> m;
    
    int arr[n];
    for(int i=0; i<n; i++)
        cin >> arr[i];
    
    int answer = 0;
    for(int i=0; i<n; i++){
        int sum = arr[i];
        for(int j=i+1; j<n; j++){
            if( (sum+arr[j]) < m ) 
                sum += arr[j];
            else if( (sum+arr[j]) == m ) {
                answer++;
                break;
            }
            else
                break;
        }
    }
    
    cout << answer;
    return 0;
}

 

 

 

수정한 코드는 다음과 같다.

sum의 값을 선언하는 부분을 수정했다.

내부의 for문을 i+1부터 말고 i부터 돌도록 수정했더니 잘 돌아갔다.

#include <iostream>
using namespace std;

int main(){
    int n,m;
    cin >> n >> m;
    
    int arr[n];
    for(int i=0; i<n; i++)
        cin >> arr[i];
    
    int answer = 0;
    for(int i=0; i<n; i++){
        int sum = 0;
        for(int j=i; j<n; j++){
            if( (sum+arr[j]) < m ) 
                sum += arr[j];
            else if( (sum+arr[j]) == m ) {
                answer++;
                break;
            }
            else
                break;
        }
    }
    
    cout << answer;
    return 0;
}