• C. Planning(贪心)


    C. Planning
    time limit per test
    1 second
    memory limit per test
    512 megabytes
    input
    standard input
    output
    standard output

    Helen works in Metropolis airport. She is responsible for creating a departure schedule. There are n flights that must depart today, the i-th of them is planned to depart at the i-th minute of the day.

    Metropolis airport is the main transport hub of Metropolia, so it is difficult to keep the schedule intact. This is exactly the case today: because of technical issues, no flights were able to depart during the first k minutes of the day, so now the new departure schedule must be created.

    All n scheduled flights must now depart at different minutes between (k + 1)-th and (k + n)-th, inclusive. However, it's not mandatory for the flights to depart in the same order they were initially scheduled to do so — their order in the new schedule can be different. There is only one restriction: no flight is allowed to depart earlier than it was supposed to depart in the initial schedule.

    Helen knows that each minute of delay of the i-th flight costs airport ci burles. Help her find the order for flights to depart in the new schedule that minimizes the total cost for the airport.

    Input

    The first line contains two integers n and k (1 ≤ k ≤ n ≤ 300 000), here n is the number of flights, and k is the number of minutes in the beginning of the day that the flights did not depart.

    The second line contains n integers c1, c2, ..., cn (1 ≤ ci ≤ 107), here ci is the cost of delaying the i-th flight for one minute.

    Output

    The first line must contain the minimum possible total cost of delaying the flights.

    The second line must contain n different integers t1, t2, ..., tn (k + 1 ≤ ti ≤ k + n), here ti is the minute when the i-th flight must depart. If there are several optimal schedules, print any of them.

    Example
    input
    5 2
    4 2 1 10 2
    output
    20
    3 6 7 4 5
    Note

    Let us consider sample test. If Helen just moves all flights 2 minutes later preserving the order, the total cost of delaying the flights would be (3 - 1)·4 + (4 - 2)·2 + (5 - 3)·1 + (6 - 4)·10 + (7 - 5)·2 = 38 burles.

    However, the better schedule is shown in the sample answer, its cost is (3 - 1)·4 + (6 - 2)·2 + (7 - 3)·1 + (4 - 4)·10 + (5 - 5)·2 = 20burles.

    算法:贪心

    注意:不要用vector.erase,时间复杂度是O(n),用set.erase,时间复杂度时O(n)。

    #include <iostream>
    #include <cstdio>
    #include <queue>
    #include <algorithm>
    #include <set>
    
    using namespace std;
    
    const int maxn = 1e7+7;
    
    typedef long long ll;
    
    struct node {
        ll id;
        ll val;
    }arr[maxn];
    
    set<ll> vis;
    long long ans[maxn];
    int n, k;
    
    bool cmp(node x, node y) {
        if(x.val == y.val) {
            return x.id > y.id;
        }
        return x.val > y.val;
    }
    
    int main() {
        scanf("%d %d", &n, &k);
        for(int i = 1; i <= n; i++) {
            scanf("%lld", &arr[i].val);
            arr[i].id = i;
            vis.insert(i + k);
        }
        sort(arr + 1, arr + n + 1, cmp);
        long long sum = 0;
        for(int i = 1; i <= n; i++) {
            ll u = *vis.lower_bound(arr[i].id);
            sum += (u - arr[i].id) * arr[i].val;
            ans[arr[i].id] = u;
            vis.erase(u);  
        }
        printf("%lld
    ", sum);
        for(int i = 1; i <= n; i++) {
            printf("%lld ", ans[i]);
        }
        return 0;
    }
  • 相关阅读:
    分享关于Entity Framework 进行CRUD操作实验的结果
    总结Unity IOC容器通过配置实现类型映射的几种基本使用方法
    Python深入:Distutils发布Python模块--转载
    原创:R包制作--windows
    提高R语言速度--转载
    R 语言 Windows 环境 安装与Windows下制作R的package--Rtools
    极简 R 包建立方法--转载
    R的极客理想系列文章--转载
    如何创建R包并将其发布在 CRAN / GitHub 上--转载
    教你如何成为数据科学家(六)
  • 原文地址:https://www.cnblogs.com/buhuiflydepig/p/11329858.html
Copyright © 2020-2023  润新知