4655. 重新排序

给定一个数组 A 和一些查询 Li,Ri,求数组中第 Li 至第 Ri 个元素之和。

小蓝觉得这个问题很无聊,于是他想重新排列一下数组,使得最终每个查询结果的和尽可能地大。

小蓝想知道相比原数组,所有查询结果的总和最多可以增加多少?

输入格式
输入第一行包含一个整数 n。

第二行包含 n 个整数 A1,A2,···,An,相邻两个整数之间用一个空格分隔。

第三行包含一个整数 m 表示查询的数目。

接下来 m 行,每行包含两个整数 Li、Ri,相邻两个整数之间用一个空格分隔。

输出格式
输出一行包含一个整数表示答案。

数据范围
对于 30% 的评测用例,n,m≤50;
对于 50% 的评测用例,n,m≤500;
对于 70% 的评测用例,n,m≤5000;
对于所有评测用例,1≤n,m≤105,1≤Ai≤106,1≤Li≤Ri≤n。

输入样例:
5
1 2 3 4 5
2
1 3
2 5
输出样例:
4
样例解释
原来的和为 6+14=20,重新排列为 (1,4,5,2,3) 后和为 10+14=24,增加了 4。

思路 :

  • 显然贪心思路就是让出现次数越多的位置上的数越大
  • 由于只需要修改不需要查询,因此我们可以通过 一维差分 统计每个位置的次数,对差分做前缀和即可得到每个位置出现次数的数组
  • 原先查询结果的总和 = 每个位置出现次数 * 位置上的数
  • 排列后查询结果的总和 = 每个位置出现次数 * 重新排列后位置上对应的数,由于数的大小与次数成正比,因此只要分别对两个数组排序即可
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;

int n, m;
int a[N], cnt[N];

int main() {
    cin >> n;
    for (int i = 1; i <= n; ++ i) cin >> a[i];
    cin >> m;
    ll sum1 = 0, sum2 = 0;
    for (int i = 1, l, r; i <= m; ++ i) {
        cin >> l >> r;
        cnt[l] ++ ;
        cnt[r + 1] -- ;
    }
    for (int i = 1; i <= n; ++ i) {
        cnt[i] += cnt[i - 1];
        sum1 += (ll)a[i] * cnt[i];
    }
    sort(a + 1, a + n + 1);
    sort(cnt + 1, cnt + n + 1);
    for (int i = 1; i <= n; ++ i) {
        sum2 += (ll)a[i] * cnt[i];
    }
    cout << sum2 - sum1;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值