acwing4655.重新排序

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

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

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

输入格式

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

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

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

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

输出格式

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

数据范围

对于 30%30% 的评测用例,n,m≤50;
对于 50%50% 的评测用例,n,m≤500
对于 70%70% 的评测用例,n,m≤5000
对于所有评测用例,1≤n,m≤10^5,1≤Ai≤10^6,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<cstring>
#include<algorithm>
using namespace std;
const int maxn=1e6+10;
typedef long long LL;
int a[maxn],b[maxn],n,m,arr[maxn];
LL res=0,ans=0,cnt=0;
LL quickcount(int a,int b)
{
    LL sum=0;
    for(int i=a;i<=b;i++)
    {
        sum+=arr[i];
    }
    return sum;
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>arr[i];
    }
    cin>>m;
    for(int i=0;i<m;i++)
    {
        cin>>a[i]>>b[i];
        cnt+=quickcount(a[i],b[i]);
    }
    sort(arr+1,arr+n+1);
    do
    {
        for(int i=0;i<m;i++)
        {
            ans+=quickcount(a[i],b[i]);
        }
        res=max(ans,res);
        ans=0;
    }while(next_permutation(arr+1,arr+n+1));            //    全排列取最大值res;
    cout<<res-cnt<<endl;
}

对于一般较大的数组,这样肯定是会超时的;所以需要考虑到算法优化的问题。

看了讲解,有不懂的可以上acwing基础算法课上学习差分,当Ai数组为正序,且从Ci数组也为正序的时候,ai*ci的和最大(ci数组为下标i对应ai数组需要加的次数)

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1e6+10;
typedef long long LL;
int w[maxn],s[maxn];                w数组存放数字,s数组存放每个数字需要计算的次数
int n,m;
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>w[i];
    }
    cin>>m;
    while(m--)
    {
        int l,r;
        cin>>l>>r;
        s[l]++,s[r+1]--;
    }
    for(int i=1;i<=n;i++)
    {
        s[i]+=s[i-1];
    }
    LL sum1=0;
    for(int i=1;i<=n;i++)
    {
        sum1+=(LL)w[i]*s[i];
    }
    sort(w+1,w+n+1);                //将w,s数组都正序排列,计算res最大值。
    sort(s+1,s+n+1);
    LL sum2=0;
    for(int i=1;i<=n;i++)
    {
        sum2+=(LL)w[i]*s[i];
    }
    cout<<sum2-sum1<<endl;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值