算法笔记练习 4.5 二分 问题 B: 打印极值点下标

算法笔记练习 题解合集

题目链接

题目

题目描述
在一个整数数组上，对于下标为i的整数，如果它大于所有它相邻的整数，或者小于所有它相邻的整数，则称为该整数为一个极值点，极值点的下标就是i。

输入
每个案例的输入如下：

有2×n+1行输入：第一行是要处理的数组的个数n；
对其余2×n行，第一行是此数组的元素个数k(4<k<80)，第二行是k个整数，每两个整数之间用空格分隔。

输出
每个案例输出不多于n行：每行对应于相应数组的所有极值点下标值，下标值之间用空格分隔，如果没有极值点则不输出任何东西。

样例输入 Copy
2
4
1 2 1 3
5
3 4 5 6 7
样例输出 Copy
0 1 2 3
0 4

思路

本题代码核心是 print_extreme 函数，它接收长度为 n 的数组 nums，并打印极值点下标。该函数的思路如下：

1. 新建一个数组 extremes[n]，将其所有元素初始化为 0，extremes[i] == 1 代表 nums[i] 是极值点
2. 分别判断第一个点、中间的点、最后的点是否为极值点，如果是的话将 extremes 数组中对应的元素改为 1
3. 根据数组 extremes 来输出结果

细节：

1. 长度为 n 的数组，最后一个元素是 nums[n-1]
2. 每行行末不能有空格
3. 没有极值点的时候不能输出任何东西，包括换行

代码

#include <stdio.h>
// print_extreme接收长度为n的数组nums，打印极值点下标 
void print_extreme(int n, int nums[n]); 
int main(){
	int n, k, i;
	while (scanf("%d", &n) != EOF){
		while (n--){
			scanf("%d", &k);
			int input[k];
			for (i = 0; i < k; ++i)
				scanf("%d", &input[i]);
			print_extreme(k, input); 
		} 
	} 
	return 0;
}

void print_extreme(int n, int nums[n]){
	// 分两边和中间处理 
	// 若nums[i]是极值点，extremes[i]设为1 
	int i;
	int extremes[n];
	for (i = 0; i < n; ++i)
		extremes[i] = 0;
	// 判断nums[0]是否为极值点 
	if (nums[0] != nums[1])
		extremes[0] = 1;
	// 判断nums[1]到nums[n-2]是否为极值点 
	for (i = 1; i < n-1; ++i){
		if( (nums[i] < nums[i-1] && nums[i] < nums[i+1])
				|| (nums[i] > nums[i-1] && nums[i] > nums[i+1]))
			extremes[i] = 1;
	}
	// 判断nums[n-1]是否为极值点
	if (nums[n-1] != nums[n-2])
		extremes[n-1] = 1;
	// flag用来控制格式，0表示第一个极值点还未输出，1则相反 
	int flag = 0;
	for (i = 0; i < n; ++i){
		if (extremes[i]){
			if (flag)
				putchar(' ');
			printf("%d", i);
			flag = 1;
		} 
	}
	// 注意！仅当有极值点才换行 
	if (flag)
		putchar('\n');
	return;
}