Codeforces Round #433 (Div. 2) C. Planning(贪心)

题目链接：Codeforces Round #433 (Div. 2) C. Planning

题意：

有n架飞机要起飞，起飞的时间是t₁=k+1,t₂=k+2...t_n=k+n。

现在有n个位置，每个位置的cost为(t-i)*a_i(t>=i),现在问你如何给每架飞机安排位置，使得sumcost最小。

题解：

考虑贪心，先将a_i 从大到小排序，显然对于大的a，乘的数越小越好，然后用set搞一搞就行了。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=a;i<=b;++i)
using namespace std;
typedef long long ll;
typedef pair<int,int>P;

const int N=3e5+7;
multiset<int>st;
int n,k,ans[N];
P a[N];
ll ret=0;
int main(){
    scanf("%d%d",&n,&k);
    F(i,1,n)scanf("%d",&a[i].first),a[i].second=i;
    sort(a+1,a+1+n,greater<P>());
    F(i,1,n)st.insert(k+i);
    F(i,1,n)
    {
        auto it=st.lower_bound(a[i].second);
        ans[a[i].second]=*it;
        ret+=(ll)(*it-a[i].second)*a[i].first;
        st.erase(it);
    }
    printf("%lld\n",ret);
    F(i,1,n)printf("%d%c",ans[i]," \n"[i==n]);
    return 0;
}