2019 CAL Season 1 #F 풀이

출제자 : 31기 이온조

정답 코드

#include <bits/stdc++.h>
using namespace std;
 
int main() {
    long long ans = 0;
    int N; scanf("%d",&N);
    for(int i=1; i<=N; i++) {
        int foo; scanf("%d",&foo);
        ans += foo;
    }
    printf("%lld", ans * ans * ans);
    return 0;
}
 

 

풀이법

문제에 주어진 식을 3중 반복문으로 계산한다면 시간 내에 문제를 해결할 수 없습니다.

조금만 고민해 본다면, 문제에서 주어진 식을 다음과 같이 변형할 수 있다는 것을 알 수 있습니다.

오른쪽 식을 전개한다고 생각해보면 왼쪽과 같이 됨을 알 수 있다

이제 수열 A의 합을 구한 다음, 이를 세제곱하여 출력하면 답을 구할 수 있습니다.