Codeforces Round #538 (Div. 2) B. Yet Another Array Partitioning Task

最新推荐文章于 2022-05-15 16:28:17 发布

萧瑟1

最新推荐文章于 2022-05-15 16:28:17 发布

阅读量216

点赞数

CC 4.0 BY-SA版权

分类专栏： codeforces

本文链接：https://blog.youkuaiyun.com/qq_41410799/article/details/86983502

codeforces 专栏收录该内容

54 篇文章

订阅专栏

本文介绍了一种通过排序和使用map数据结构来快速解决区间求和问题的算法。首先，将数组从大到小排序，然后累加前m*k个元素得到总和。对于特定的区间求和需求，利用预先存储的map进行高效查询。

题目链接：

传送门

思路：

先从大到小排序，然后将前m*k个数相加就是答案。

对于求区间，我是用的map存的，相对比较慢一些，但是也比较好想。

代码如下：

#include <bits/stdc++.h>
using namespace std;
const int maxn=2*1e5+5;
map<int,int>ma;
typedef long long ll;
int n,m,k;
int a[maxn];
int b[maxn];
int no[maxn];
ll ans=0;
int compare (int a,int b)
{
    return a>b;
}
int main ()
{
    scanf("%d%d%d",&n,&m,&k);
    for (int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
        b[i]=a[i];
    }
    sort(b,b+n,compare);
    for (int i=0;i<m*k;i++)
    {
        ans+=b[i];
        ma[b[i]]++;
    }
    printf("%lld\n",ans);
    int cnt=0;
    for (int i=0,j=0;i<k-1;i++)
    {
        int t=0;
        while (t<m)
        {
            if(ma[a[j]])
            {
                ma[a[j]]--;
                t++;
            }
            j++;
        }
        no[cnt++]=j;
    }
    for (int i=0;i<cnt;i++)
        printf("%d ",no[i]);
    return 0;
}