백준 14247번 문제 ( 나무 자르기 )

문제 출처 : https://www.acmicpc.net/problem/14247

14247번: 나무 자르기

이 문제는 현재 나무들의 높이와 가중치가 주어지는데 탐욕법을 이용하여서 가중치 순으로 정렬을 한 뒤

가중치가 작은 순서대로 자르면 답이 나오는 문제이다.

아래는 해당 문제를 풀이한 소스 코드이다. 

#include<stdio.h>
#include<algorithm>
#pragma warning(disable:4996)
using namespace std;
struct list
{
    int tree_height;
    int increase;
};
list arr[100001];
bool compare(const list & a, const list & b)
{
    return a.increase < b.increase;
}
int main()
{
    int n;
    long long int value = 0;
    scanf("%d", &n);
    for (int i = 0; i < n; i++)
        scanf("%d", &arr[i].tree_height);
    for (int i = 0; i < n; i++)
        scanf("%d", &arr[i].increase);
    sort(arr, arr + n, compare);
    for (int i = 0; i < n; i++)
        value = value+ arr[i].tree_height + i * arr[i].increase;
    printf("%lld", value);
    
}