【BCPC2016 NAIVE排序】快读 | STL | Trie | E

【BCPC 2016】NAIVE排序

时间限制: 1000 ms 内存限制: 131072 kb 总通过人数: （未加载） 总提交人数: （未加载）

Tags： STL Trie

题目描述

给定 $n$ 行字符串，每行字符串的长度不定 ，请按字典序对它们进行排序并输出第 $k$ 小（从 $1$ 开始数）的字符串。

输入

$T$ 组（$T\le 20$）

对于每组数据，包含若干行，第一行两个正整数 $k$ 和 $n$

接下来 $n$ 行每行一个由 ASCII码在33到123之间 的字符组成的字符串

保证输入数据大小不超过 30MB。

范围：$1\le T\le 20，1\le N\le M\le 2.5\times 10^6$，且所有数据的字符串长度之和不超过 $2.5\times 10^7$

输出

对于每组数据输出一行一个字符串，即第 $k$ 小的字符串

输入样例

3
1 5
a
aa
aaAaa
aabaa
baa
3 5
a
aa
aaAaa
aabaa
baa
5 5
a
aa
aaAaa
aabaa
baa

输出样例

a
aaAaa
baa

分析

单纯的第 $k$ 大，首先想到的是 std::nth_element。但有个问题是字符串的长度都是不定的，且数据量还挺大。

因此不妨手动维护一个缓冲区来进行读入。然后就在缓存数组里进行字符串切片，并且把每个字符串的首指针拿出来放到一个数组里面去，再自定义 $pred$（可用 $lambda$ 表达式） 对这个数组进行 std::nth_element 就可以了。

不过这种解法应该不是出题人的本意。出题人的本意应该是想让我们利用 Trie 的思想把字符串进行归类，然后依次筛除那些不会进入前 $k$ 小的串… 之后再详细实现此解法。

时间复杂度：

• 读入 + 得到指针数组， $O(\sum _{i=1}^{n}le{n}_i)$ 的。
• $std::nth\_element$ ，近似 $O(\sum _{i=1}^{n}le{n}_i)$ 的。
• 总的 $O(\sum _{i=1}^{n}le{n}_i)$ 的。

AC代码

#include <stdio.h>
#include <string.h>
#include <algorithm>

constexpr int SZ(30000002), MN(3000006);


class Scanner
{
private:

	char buf[SZ], *p_now;

	inline bool isd(const char c)
	{
		return c <= '9' && c >= '0';
	}

	inline bool isc(const char c)
	{
		return c >= 33 && c <= 123;
	}

public:

	Scanner(void) :
		p_now(buf) { }

	void read(void)
	{
		fread(buf, 1, SZ, stdin);
	}

	int nextInt(void)
	{
		int ret = 0;

		while (!isd(*p_now))
			++p_now;
		do ret = ret * 10 + *p_now++ - '0';
		while (isd(*p_now));

		return ret;
	}

	char *nextLine(void)
	{
		while (!isc(*p_now))
			++p_now;
		char *begin = p_now;
		while (isc(*p_now))
			++p_now;
		*p_now = '\0';
	
		return begin;
	}
};


Scanner in;
char *str[MN];

int main()
{
	in.read();

	int T = in.nextInt();
	int k, n;
	while (T--)
	{
		k = in.nextInt(), n = in.nextInt();
		for (int i=0; i<n; ++i)
			str[i] = in.nextLine();

		std::nth_element(str, str+k-1, str+n,
			[](const char *p, const char *q) -> bool
			{
				return strcmp(p, q) < 0;
			}
		);
		puts(str[k-1]);
	}

	return 0;
}

BCPC2018 + 2019选拔赛要来了，加油鸭！