题目:就是找出所有的区间的前k大的数。
最近写题状态不好,简单的题目也花费很长的时间,加油叭。
错误的思路:这是我开始想题目的思路,题目给出的n是1e5所以直接去算所有的区间和的话是不可能的,所以我很正常的想到了只能求出前k个大的,但是接下来的思路却出现了问题,我们知道最大的肯定就是所有的数加在一起,那第二大的呢,第二大的肯定就是所有的数加在一起然后减去第一个或者减去最后一个,到这里我的思路还是没错很多,但是接下来的第三大的呢,我考虑到第三大的可能是第一大的减去第一个或者最后一个,也可能是第二大的减去最左边的一个或者最右边的一个,所以我根本就不能继续写下去,然后我的思路就开始混乱。
正确的思路: 其实题目非常的简单,因为我们实际上要求的就是所有区间的前k大的数,所以我们只需要将所有区间可能的最大的情况放在一个优先队列进行比较然后输出最大的之后再将最大的进行改变就行,也就是说以i为有顶点的所有的情况中的最大情况加入我们的优先队列中。
代码:
#include<iostream>
#include<queue>
#include<map>
using namespace std;
const int N = 1e5+5;
long long a[N];
struct node
{
int l , r;
long long num ;
bool operator <(const node w)const
{
return w.num > num;
}
};
priority_queue<node> q;int main()
{
int n , k ;
cin >> n >> k ;
for(int i = 1 ; i <= n ; i ++)
scanf("%lld",&a[i]),a[i] += a[i - 1];
for(int i = 1 ; i<= n ; i ++)
q.push({1,i,a[i]});
while(k)
{
node t = q.top();
q.pop();
printf("%lld ",t.num);
t.l ++ ;
q.push({t.l,t.r,a[t.r] - a[t.l-1]});
k--;
}
return 0;
}