4655. 重新排序差分排序不等式贪心

最新推荐文章于 2024-08-21 11:22:17 发布

我也念过晚霞

最新推荐文章于 2024-08-21 11:22:17 发布

阅读量153

点赞数 3

分类专栏： AcWing 文章标签：算法数据结构

本文链接：https://blog.csdn.net/2303_79132118/article/details/136437760

版权

AcWing 专栏收录该内容

239 篇文章 0 订阅

订阅专栏

本文介绍了一个C++程序，利用差分数组和排序技巧解决计算有序数组中元素与索引乘积问题，以提升计算效率。

摘要由CSDN通过智能技术生成

#include <iostream>
#include <algorithm>
#define N 100005
using namespace std;

int n, m, l, r, a[N], b[N];
long long sum, ans;

int main ()
{
    cin >> n;
    for (int i = 1; i <= n; i ++)
    {
        cin >> a[i];
    }
    cin >> m;
    while (m --)
    {
        cin >> l >> r, b[l] ++, b[r + 1] --; // 同时处理差分
    }
    for (int i = 1; i <= n; i ++)
    {
        b[i] += b[i - 1], sum += (long long) a[i] * b[i]; // 先得到数组 c（仍存放在数组 b 中），在顺便统计原来的总和 sum
    }
    sort (a + 1, a + n + 1), sort (b + 1, b + n + 1); // 分别排序
    for (int i = 1; i <= n; i ++)
    {
        ans += (long long) a[i] * b[i]; // 加上依次相乘的和
    }
    cout << ans - sum; // 答案即是依次相乘总和 - sum
    return 0;
}