[Softeer] 성적평균 (C++)

📝 제출 코드 #1

#include<iostream>
#include<vector>

using namespace std;

int main(int argc, char** argv)
{
    int N, K; // N: 학생 수, K: 구간 수
    vector<float> arr;

    cin >> N >> K;
    
    arr.push_back(0);
    
    for(int i=0; i<N; i++){
        float num;
        cin >> num;
        arr.push_back(num);
    }

    for(int i=0; i<K; i++){
        int A,B;

        cin >> A >> B;
        float sum=0;
        for(int j=A; j<=B; j++){
            sum += arr[j];
        }
        printf("%.2f\n", sum/(B-A+1));
        
    }
    
    return 0;
}

 

"차이가 0.01이하이면 정답으로 채점됨"이라고 해서 그냥 나눗셈으로 계산하고, 소수점 둘째자리까지 출력되도록 함

 

 

📝 제출 코드 #2

#include<iostream>
#include<vector>

using namespace std;
int main(int argc, char** argv)
{
    int N, K; // N: 학생 수, K: 구간 수
    vector<float> sum;

    cin >> N >> K;
    
    sum.push_back(0);
    
    for(int i=0; i<N; i++){
        float num;
        cin >> num;
        sum.push_back(sum[i]+num);
    }

    for(int i=0; i<K; i++){
        int A,B;

        cin >> A >> B;
        
        printf("%.2f\n", (sum[B]-sum[A-1])/(B-A+1));
        
    }
    
    return 0;
}

구간합 알고리즘(누적합 알고리즘)으로 풀이한 코드이다.

반복문을 1번만 사용하기 때문에 복잡도가 더 줄었다.

누적합 넣을 때 sum[i+1] = sum[i]+num했을 때 해당 예제로는 통과하는데 제출 시 다른 TC들은 런타임에러가 발생해서, push_back 쓰는걸로 바꿨다.