Week2_2041干草堆(差分/前缀和)

学到了一个新的STL:nth_element
位于头文件algorithm中,这个相当于是算法课本第二章的线性时间选择,在线性时间内选择出第k大/小个元素,非常的简单
主要的原理是把第k个位置的交换一下,所以最后输出arr[k]就好
用法

// 选出第k个元素
nth_element(arr+1,arr+k,arr+n+1);
nth_element(arr.begin(),arr.begin()+k,arr.end(),cmp);

// 输出结果
cout<<arr[k];

开始解题

题目链接

这个题目运用到了差分的知识点,模板题

前缀和

如果想求数组arr[l,r]之间的和,可以引入一个数组ed[r]表示的是arr[1,r]的值,即
ed[i] = \begin{cases} arr[1] & i=1 \\ ed[i-1]+arr[i] &else \end{cases}
就可以将问题转换为ed[r]-ed[l-1]

差分

长度为 n 的整数序列,m 个操作,每个操作包含三个整数 l,r,c,表示将序列中 [l,r] 之间的每个数加上 c。输出进行完所有操作后的序列。

解法:给区间[l, r]中的每个数加上c:
B[l] += c, B[r + 1] -= c

#include<bits/stdc++.h>
using namespace std;
int n,m,l,r,c;
int arr[100005];
int ed[100005];
int main(){
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&arr[i]);
        ed[i]=arr[i]-arr[i-1];
    }
    while(m--){
        scanf("%d%d%d",&l,&r,&c);
        ed[l]+=c;
        ed[r+1]-=c;
    }
    for(int i=1;i<=n;i++){
        ed[i]+=ed[i-1];
        cout<<ed[i]<<" ";
    }
    return 0;
}

本题解法

所以就只需要计算套用差分模板

#include<iostream>
#include<algorithm>
using namespace std;
int N,K,A,B;
int arr[1000005];
int main(){
    cin>>N>>K;
    for(int i=1;i<=K;i++){
        cin>>A>>B;
        arr[A]++;
        arr[B+1]--;
    }
    for(int i=1;i<=N;i++){
        arr[i]+=arr[i-1];
    }
    nth_element(arr+1,arr+(N+1)/2,arr+N+1);
    cout<<arr[(N+1)/2];
    return 0;
}
image.png
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容