2041 干草堆

最新推荐文章于 2022-08-11 21:02:29 发布

三无提督w

最新推荐文章于 2022-08-11 21:02:29 发布

阅读量118

点赞数

分类专栏： # Acwing 文章标签：算法 c++ python

本文链接：https://blog.csdn.net/cf260469080/article/details/122388740

版权

Acwing 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

2041 干草堆

贝茜对她最近在农场周围造成的一切恶作剧感到抱歉，她同意帮助农夫约翰把一批新到的干草捆堆起来。

开始时，共有 N 个空干草堆，编号 1∼N。

约翰给贝茜下达了 K 个指令，每条指令的格式为 A B，这意味着贝茜要在 A…B 范围内的每个干草堆的顶部添加一个新的干草捆。

例如，如果贝茜收到指令 10 13，则她应在干草堆 10,11,12,13 中各添加一个干草捆。

在贝茜完成了所有指令后，约翰想知道 N 个干草堆的中值高度——也就是说，如果干草堆按照高度从小到大排列，位于中间的干草堆的高度。

方便起见，N 一定是奇数，所以中间堆是唯一的。

请帮助贝茜确定约翰问题的答案。

输入格式

第一行包含 N 和 K。

接下来 K 行，每行包含两个整数 A,B，用来描述一个指令。

输出格式

输出完成所有指令后，N 个干草堆的中值高度。

数据范围

$1≤N≤10^6,\\ 1≤K≤25000,\\ 1≤A≤B≤N$

输入样例：

输出样例：

样例解释

贝茜完成所有指令后，各堆高度为 0,1,2,3,3,1,0。

将各高度从小到大排序后，得到 0,0,1,1,2,3,3，位于中间的是 1。

题解

python（差分）

4543 ms

n, k = map(eval, input().split())
b=[0 for i in range(n+1)]
ls=[]
for i in range(k):
    A, B = map(eval, input().split())
    b[A-1]+=1
    b[B]-=1

ls.append(b[0])
for i in range(1,n):
    ls.append(b[i]+ls[i-1])
ls.sort()
print(ls[n//2])

python（暴力枚举，会TLE）

n, k = map(eval, input().split())
ls=[0 for i in range(n)]
for i in range(k):
    A, B = map(eval, input().split())
    for j in range(A-1,B):
        ls[j]+=1
ls.sort()
print(ls[n//2])

c++（差分）

对于sort行，可以用nth_element进行优化，优化后从849 ms到230 ms

#include <cstdio>
#include <algorithm>

using namespace std;

int n, k, b[1000000];
int main() {

    scanf("%d%d", &n, &k);
    while (k--) {
        int A, B;
        scanf("%d%d", &A, &B);
        b[A - 1]++, b[B]--;
    }
    for (int i = 0; i < n; i++) {
        b[i + 1] += b[i] ;
    };
    sort(b,b+n);
    // nth_element(b,b+n/2,b+n);
    printf("%d\n",b[n/2]);
    return 0;
}