C++/코드트리

[코드트리] 원소 값들의 최대 합

sondiaa 2023. 8. 18. 00:19

https://www.codetree.ai/missions/5/problems/maximum-sum-of-element-values?utm_source=clipboard&utm_medium=text 

1번인덱스부터 탐색을 하도록 했고, 모두 m번 탐색시켜서 값을 구하겠다.

#include <iostream>
#include <algorithm>
using namespace std;
int n, m, map[101];
int main() {
    cin >> n >> m;
    for(int i=1; i<=n; i++)
        cin >> map[i];
    int answer = 0;
    for(int i=1; i<=n; i++){
        int sum = 0;
        int startIdx = i;
        for(int j=0; j<m; j++){
            sum += map[startIdx];
            startIdx = map[startIdx];           
        }

        answer = max(answer, sum);
    }

    cout << answer;
    return 0;
}