【数据结构】——解决topk问题

前言:我们前面已经学习了小堆并且也实现了小堆,那么我们如果要从多个数据里选出最大的几个数据该怎么办呢,这节课我们就来解决这个问题。我们就用建小堆的方法来解决。

在这里插入图片描述

首先我们来看到这个方法的时间复杂度,我们先取前k个数据建立一个小堆,后面插入的数据依次与堆顶对比,比堆顶小就不进入堆,比堆顶大就代替堆顶进入堆,在进行向下调整。时间复杂度为O(N*logK)。

创造随机数,存入文件:

void CreateNDate()
{
	// 造数据
	int n = 10000000;
	srand(time(0));
	const char* file = "data.txt";
	FILE* fin = fopen(file, "w");
	if (fin == NULL)
	{
		perror("fopen error");
		return;
	}

	for (int i = 0; i < n; ++i)
	{
		int x = (rand() + i) % 10000000;
		fprintf(fin, "%d\n", x);
	}

	fclose(fin);
}

我们设置1000万个1000万以内的数,并且将它存入文件。
在这里插入图片描述
我们打开文件的路径在点开文件就可以找到我们生成的随机数了。

我们来看看如何建堆:

void PrintTopK(const char* file, int k)
{
	FILE* fout = fopen(file, "r");
	if (fout == NULL)
	{
		perror("fopen error");
		return;
	}

	// 建一个k个数小堆
	int* minheap = (int*)malloc(sizeof(int) * k);
	if (minheap == NULL)
	{
		perror("malloc error");
		return;
	}

	// 读取前k个,建小堆
	for (int i = 0; i < k; i++)
	{
		fscanf(fout, "%d", &minheap[i]);
		AdjustUp(minheap, i);
	}

	int x = 0;
	while (fscanf(fout, "%d", &x) != EOF)
	{
		if (x > minheap[0])
		{
			minheap[0] = x;
			AdjustDown(minheap, k, 0);
		}
	}

	for (int i = 0; i < k; i++)
	{
		printf("%d ", minheap[i]);
	}
	printf("\n");

	free(minheap);
	fclose(fout);
}

因为我们取前k个数据来建立一个小堆,我们比堆顶大的数据就代替堆顶在向下调整沉到堆底。我们再将堆依次从文件中读取出来,如果我们读取出来的数据大于堆顶元素,就代替堆顶进行向下调整,最后在打印堆就可以打印出来前k个大的数据了。

测试代码:

int main()
{
	 CreateNDate();
	PrintTopK("Data.txt", 5);

	return 0;
}

在这里插入图片描述

如果我们在文件中改入几个大于1000万的数进行测试:
在这里插入图片描述

那么我们放入的数据就打印出来了。最后感谢大家的支持!

  • 46
    点赞
  • 44
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 38
    评论
下面是一个用C语言实现迷宫问题的完整代码。该程序可以读取迷宫地图文件,然后使用深度优先搜索算法找出从起点到终点的最短路径。 ``` #include <stdio.h> #include <stdlib.h> #define ROW 10 #define COL 10 int maze[ROW][COL]; // 迷宫地图 int visited[ROW][COL]; // 记录是否访问过 int path[ROW * COL][2]; // 保存路径 int top = 0; // 栈顶指针 int load_map(char *filename); // 读取地图文件 void print_map(); // 打印地图 void dfs(int x, int y, int step); // 深度优先搜索 int main() { int start_x, start_y, end_x, end_y, i; // 读取地图文件 if (load_map("map.txt") == 0) { printf("Failed to load map.\n"); return -1; } // 打印地图 print_map(); // 输入起点和终点的坐标 printf("Please input the start point (x, y): "); scanf("%d %d", &start_x, &start_y); printf("Please input the end point (x, y): "); scanf("%d %d", &end_x, &end_y); // 开始搜索 dfs(start_x, start_y, 0); // 打印路径 printf("The path from (%d,%d) to (%d,%d) is:\n", start_x, start_y, end_x, end_y); for (i = 0; i < top; i++) { printf("(%d,%d) ", path[i][0], path[i][1]); } printf("\n"); return 0; } // 读取地图文件 int load_map(char *filename) { FILE *fp; int i, j; fp = fopen(filename, "r"); if (fp == NULL) { return 0; } for (i = 0; i < ROW; i++) { for (j = 0; j < COL; j++) { fscanf(fp, "%d", &maze[i][j]); visited[i][j] = 0; } } fclose(fp); return 1; } // 打印地图 void print_map() { int i, j; printf("The map is:\n"); for (i = 0; i < ROW; i++) { for (j = 0; j < COL; j++) { printf("%d ", maze[i][j]); } printf("\n"); } } // 深度优先搜索 void dfs(int x, int y, int step) { int i, j, k; path[step][0] = x; path[step][1] = y; if (x == end_x && y == end_y) { // 找到终点 top = step + 1; return; } visited[x][y] = 1; // 向上走 i = x - 1; j = y; if (i >= 0 && maze[i][j] == 0 && visited[i][j] == 0) { dfs(i, j, step + 1); if (top > 0) { return; } } // 向右走 i = x; j = y + 1; if (j < COL && maze[i][j] == 0 && visited[i][j] == 0) { dfs(i, j, step + 1); if (top > 0) { return; } } // 向下走 i = x + 1; j = y; if (i < ROW && maze[i][j] == 0 && visited[i][j] == 0) { dfs(i, j, step + 1); if (top > 0) { return; } } // 向左走 i = x; j = y - 1; if (j >= 0 && maze[i][j] == 0 && visited[i][j] == 0) { dfs(i, j, step + 1); if (top > 0) { return; } } visited[x][y] = 0; // 回溯 } ``` 在上面的代码中,`load_map`函数用于读取地图文件,`print_map`函数用于打印地图,`dfs`函数用于执行深度优先搜索。在`dfs`函数中,我们首先记录当前位置到路径数组中,然后分别尝试向上、向右、向下、向左走,并递归调用`dfs`函数。如果到达终点,我们就将路径保存到`path`数组中,并返回。如果没有找到终点,我们就回溯到上一个位置,并尝试其他方向。`visited`数组用于标记当前位置是否已经访问过,防止走重复的路线。 注意,在上面的代码中,我们使用了一个`path`数组来保存路径,这个数组可以看作是一个栈。当我们找到终点时,可以通过回溯路径上的点,依次将其保存到栈中,最后再依次弹出栈中的元素,就可以得到从起点到终点的路径了。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 38
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Lehjy

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值