【jzoj4815】【ksum】【set】

inklutcuah

于 2016-12-25 10:42:19 发布

阅读量431

点赞数

CC 4.0 BY-SA版权

分类专栏： jzoj 数据结构文章标签： jzoj4815 ksum

本文链接：https://blog.youkuaiyun.com/chunkitlau/article/details/52748675

jzoj 同时被 2 个专栏收录

275 篇文章

订阅专栏

数据结构

70 篇文章

订阅专栏

本文介绍了一种解决K-Sum问题的有效算法。通过记录数组的后缀并利用set进行维护，确保了处理过程的效率。该方法适用于正数数组，并能够通过逐个元素删除来覆盖所有可能的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目大意

解题思路

我们发现可以记录从第i位开始的数组的后缀，因为数为正数所以有单调性，所以我们可以逐个数从后缀中删除，这样可以考虑所有情况，用set维护就好了。

code

#include<set>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LF double
#define LL long long
#define max(n1,n2) ((n1>n2)?n1:n2)
#define min(n1,n2) ((n1>n2)?n2:n1)
#define num(n1,n2) ((n1-1)*3*n+n2+1)
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const maxn=100000,inf=2147483647;
int n,k,a[maxn+10];LL s[maxn+10];
struct rec{
    int pos;LL val;
    friend bool operator<(rec x,rec y){
        return (x.val>y.val)||((x.val==y.val)&&(x.pos<y.pos));
    }
};
multiset<rec> se;
int main(){
    freopen("ksum.in","r",stdin);
    freopen("ksum.out","w",stdout);
    scanf("%d%d",&n,&k);
    fo(i,1,n)scanf("%d",&a[i]),s[i]=s[i-1]+a[i];
    rec tmp,nex;
    fo(i,1,n){
        tmp={n,s[n]-s[i-1]};
        se.insert(tmp);
    }
    fo(i,1,k){
        tmp=*se.begin();
        printf("%lld ",tmp.val);
        se.erase(se.begin());
        if(tmp.val-a[tmp.pos]<=0)continue;
        nex={tmp.pos-1,tmp.val-a[tmp.pos]};
        se.insert(nex);
    }
    return 0;
}