蓝桥杯刷题 前缀和与差分-[2128]重新排序(C++)

问题描述

给定一个数组 A 和一些查询 L**i, R**i,求数组中第 L**i 至第 R**i 个元素之和。 小蓝觉得这个问题很无聊,于是他想重新排列一下数组,使得最终每个查询结果的和尽可能地大。小蓝想知道相比原数组,所有查询结果的总和最多可以增加多少?

输入格式

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

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

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

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

输出格式

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

样例输入

复制代码
5
1 2 3 4 5
2
1 3
2 5

样例输出

复制代码
4

样例说明

原来的和为 6 + 14 = 20,重新排列为 (1, 4, 5, 2, 3) 后和为 10 + 14 = 24,增

加了 4。

评测用例规模与约定

对于 30% 的评测用例,n, m ≤ 50 ;

对于 50% 的评测用例,n, m ≤ 500 ;

对于 70% 的评测用例,n, m ≤ 5000 ;

对于所有评测用例,1 ≤ n, m ≤ 10^5,1 ≤ A**i ≤ 10^6,1 ≤ L**i ≤ R**i ≤ 10^6 。

知识点:前缀和与差分

代码

通过90%测试样例代码

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+5;
ll a[N],b[N],cnt[N];
ll sum1,sum2;
int main() {
	ll n,m,l,r;
  cin>>n;
  for(int i=1;i<=n;i++)
  {
    cin>>a[i];
    b[i]=b[i-1]+a[i];
  }
  cin>>m;
  for(int i=1;i<=m;i++)
  {
    cin>>l>>r;
    sum1+=b[r]-b[l-1];
    for(int j=l;j<=r;j++)
    {
      cnt[j]++;
    }
  }
  sort(a+1,a+n+1);
  sort(cnt+1,cnt+n+1);
  for(int i=1;i<=n;i++)
  {
    sum2+=a[i]*cnt[i];
  }
  cout<<sum2-sum1<<endl;
	return 0;
}

通过100%测试样例代码(差分优化)

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+5;
ll a[N],b[N],cnt[N];
ll sum1,sum2;
void insert(ll l,ll r,ll c)
{
  cnt[l]+=c;
  cnt[r+1]-=c;
}
int main() {
	ll n,m,l,r;
  cin>>n;
  for(int i=1;i<=n;i++)
  {
    cin>>a[i];
    b[i]=b[i-1]+a[i];
  }
  cin>>m;
  for(int i=1;i<=m;i++)
  {
    cin>>l>>r;
    sum1+=b[r]-b[l-1];
    insert(l,r,1);
  }
  for(int i=1;i<=n;i++)
  {
    cnt[i]+=cnt[i-1];
  }
  sort(a+1,a+n+1);
  sort(cnt+1,cnt+n+1);
  for(int i=1;i<=n;i++)
  {
    sum2+=a[i]*cnt[i];
  }
  cout<<sum2-sum1<<endl;
	return 0;
}
相关推荐
运器12322 分钟前
【一起来学AI大模型】支持向量机(SVM):核心算法深度解析
大数据·人工智能·算法·机器学习·支持向量机·ai·ai编程
Zedthm31 分钟前
LeetCode1004. 最大连续1的个数 III
java·算法·leetcode
神的孩子都在歌唱1 小时前
3423. 循环数组中相邻元素的最大差值 — day97
java·数据结构·算法
YuTaoShao1 小时前
【LeetCode 热题 100】73. 矩阵置零——(解法一)空间复杂度 O(M + N)
算法·leetcode·矩阵
眠りたいです1 小时前
Mysql常用内置函数,复合查询及内外连接
linux·数据库·c++·mysql
dying_man2 小时前
LeetCode--42.接雨水
算法·leetcode
笑鸿的学习笔记2 小时前
qt-C++语法笔记之Stretch与Spacer的关系分析
c++·笔记·qt
hardStudy_h2 小时前
C++——内联函数与Lambda表达式
开发语言·jvm·c++
vortex52 小时前
算法设计与分析 知识总结
算法
艾莉丝努力练剑3 小时前
【C语言】学习过程教训与经验杂谈:思想准备、知识回顾(三)
c语言·开发语言·数据结构·学习·算法