PTA - 新年烟花|C语言|代码详解

新年第一篇博客，祝大家新年快乐！

题目描述

新年来临，许多地方会举行烟花庆典庆祝。小 C 也想参加庆典，但不幸的是跟他一个想法的人实在太多，活动场地全是人人人人人人人人人……

活动场地可视作一个 N×M 的矩阵，其中有一些格子是空的，另外有一些格子则被人或者建筑物占领了。烟花会在一些格子上开始燃放，一个空的格子是一个对小 C 而言的优秀的观赏位置，当且仅当这个格子能看到三个及以上不同格子上燃放的烟花。一个格子能看到一个燃放的烟花，指的是：

• 格子与烟花在同一行或同一列；
• 格子与烟花的连线上的所有格子，要不是空格子，要不格子上的任何物体的高度小于小 C 的身高。

现在给定活动场地的情况以及小 C 的身高，请找到所有对小 C 而言的优秀的观赏位置，并输出能看到最多的烟花的格子的坐标。

输入格式:

输入第一行是三个正整数 N,M,H (1≤N,M≤50,1≤H≤210)，表示活动场地矩阵大小为 N×M，小 C 的身高为 H。

接下来的 N 行，每行 M 个整数，整数的含义如下：

• 如果是一个正整数，则表示该格子被人或建筑物占据，高度为对应的值。
• 如果是一个负整数，则表示该格子用于燃放烟花。所有燃放烟花的格子视为没有高度。
• 如果是 0，则表示该格子是空格子。

所有整数的绝对值不超过 500。

输出格式:

输出第一行是一个正整数，表示对小 C 而言的优秀观赏位置有多少个。

接下来输出能看到最多的燃放烟花的格子的坐标 (X,Y)，即格子在第 X 行、第 Y 列，数字间以 1 个空格分隔。当存在多组坐标时，只输出最小的那组，即输出 X 最小的解；X 相同时输出 Y 最小的解。

矩阵左上角坐标为 (0,0) ，保证至少存在一个对小 C 而言的优秀观赏位置。

输入样例:

10 10 175
0 0 0 0 0 0 0 0 0 0
0 50 0 180 -100 180 0 70 30 0
0 30 0 0 300 0 0 0 0 0
0 250 0 0 -100 0 0 0 0 0
0 -100 174 0 0 0 0 169 -100 0
0 -100 0 0 0 0 0 0 -100 0
0 -1 0 0 170 0 0 0 0 0
0 5 0 0 300 0 0 0 0 0
0 20 0 0 -100 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0

输出样例:

6

3 8

代码实现

#include <stdio.h>

// 结构体用于保存坐标信息
struct Coordinate {
    int x;
    int y;
};

// 函数声明
void find_good_place(int matrix[50][50], int row, int col, int height, struct Coordinate *good_place, int *count);

int main() 
{
    // 读入矩阵行、列和 C 的身高
    int row , col, height;
    scanf("%d %d %d", &row, &col, &height);
    
    //读入矩阵
    int i, j;
    int matrix[50][50];
    for(i = 0; i < row; i++)
    {
    	for(j = 0; j < col;j++)
    	{
    		scanf("%d", &matrix[i][j]);
    	}
    }

    struct Coordinate good_place;
    int good_place_count = 0;
	
	//调用函数 
    find_good_place(matrix, row, col, height, &good_place, &good_place_count);

	//打印结果 
    printf("%d\n", good_place_count);
    printf("%d %d\n", good_place.x, good_place.y);

    return 0;
}

void find_good_place(int matrix[50][50], int row, int col, int height, struct Coordinate *good_place, int *count) 
{
    int flag = 1, place = 0, i = 0, j = 0, k = 0, l = 0;
	//遍历矩阵上的每一个元素 
	for(i = 0; i < row; i++)
	{
		for(j = 0; j < col; j++)
		{
			place = 0;
			flag = 1; //重置flag 和 place 
			if(matrix[i][j] == 0)//该点为空位
			{
				//检查行 
				//检查所在行上是否有烟花 （条件1） 
				for(k = 0; k < col; k++)
				{
					if(matrix[i][k] < 0)
					{
						//条件1满足前提下，检查烟花点与该空位之间的建筑物高度（条件2） 
						if (k > j) // 烟花在空位的右边
						{
						    for(l = j + 1; l < k; l++) 
						    {
							    if(flag && matrix[i][l] > height)//条件2不满足 
							    {
								    flag = -1;
								    break;
							    }
						    }
						}
						else // 烟花在空位的左边
						{
						    for(l = k + 1; l < j; l++) 
						    {
							    if(flag && matrix[i][l] > height)//条件2不满足 
							    {
								    flag = -1;
								    break;
							    }
						    }
						}
						if(flag == 1)//条件2满足
						{
							place++;
						} 
					}	
				}
				//检查列 
				//检查所在列上是否有烟花 （条件1）  
				for(k = 0; k < row; k++)
				{
					flag = 1; //重置flag 
					if(matrix[k][j] < 0)
					{
						//条件1满足前提下，检查该列上的建筑物高度（条件2） 
						if (k > i) // 烟花在空位的下边
						{
						    for(l = i + 1; l < k; l++) 
						    {
							    if(flag && matrix[l][j] > height)//条件2不满足 
							    {
								    flag = -1;
								    break;
							    }
						    }
						}
						else // 烟花在空位的上边
						{
						    for(l = k + 1; l < i; l++) 
						    {
							    if(flag && matrix[l][j] > height)//条件2不满足 
							    {
								    flag = -1;
								    break;
							    }
						    }
						}
						if(flag == 1)//条件2满足
						{
							place++;
						} 
					}	
				}
			}
			//行列检查完毕，判断是否达到"这个格子能看到三个及以上不同格子上燃放的烟花"的条件 
			if (place >= 3) 
			{
		        (*count)++;
		        //第一次坐标的记录
				if (*count == 1) 
				{
		            //更新优秀观赏位置坐标
		            good_place->x = i;
		            good_place->y = j;
				}
			}
		}
	}
}

代码解析

这里讲解find_good_place函数的实现原理：

首先find_good_place函数接受6个参数，分别是活动场地的矩阵、矩阵的行大小、矩阵的列大小、小C的身高、用于记录坐标的结构体和优秀观赏位置的计数器。

在find_good_place函数中，flag用于标记空地所在行列是否有建筑物高于小C的身高，place表示空地能看到的烟花数，i、j、k、l是循环变量。

find_good_place函数是通过检查矩阵上的每一个空位（0）能看到的烟花数和烟花到空位之间的距离是否有高于小C升高的建筑物来综合判断该空位是否能成为优秀观赏位置的。具体来说代码是这样实现的：

1.首先遍历矩阵上的每一个元素是否为空位，分别检查空位所在行、列上是否有烟花。这是成为优秀观赏位置的条件之一。

2.在上一个条件满足的前提下，再检查烟花到空位之间是否有高于小C升高的建筑物。这也是成为优秀观赏位置的条件之一。

3.在上述两个条件都满足的前提下，再判断该空位上能看到不同方向上的烟花数是否大于等于3，若为真，我们可以说该空位是一个优秀观赏位置。

注意事项：

1.每一次检查完行/列准备开始检查下一个行/列时，都要重置flag的值为1，因为可能上次检查时遇到了比小C高的建筑物使flag置为-1，影响下一次检查。place同样需要在每次检查空位开始时重置为0。

2.再检查空位所在行/列上的建筑物高度时，需要判断烟花与正在检查的空位的位置关系，这影响着检查的起点与终点问题。

附：题目所给测试用例中的6个优秀观赏位置的坐标，以便调试代码：

（3，8） （4，4） （5，4） （6，8） （8，8） （9，2）