技術標籤：C語言典型例題c語言

今天來向大家介紹一個用C語言實現楊氏矩陣的問題。題目如下：

有一個數字矩陣，矩陣的每行從左到右是遞增的，矩陣從上到下是遞增的，請編寫程式在這樣的矩陣中查詢某個數字是否存在。
要求：時間複雜度小於O(N);

題幹中所描述的矩陣被稱作楊氏矩陣，然後讓你在這個這個矩陣中查詢一個數字。其實在矩陣中查詢一個數字並不難，只需採取遍歷的方式，將矩陣中每個元素拿出來比較即可。但這道題還有一個要求就是時間複雜度必須小於O（N）,也就是說不能採用遍歷的方式來查詢。因此我們需要根據楊氏矩陣的特點來寫一個新的演算法進行查詢。

如下圖所示，為一個3x3的楊氏矩陣。

根據題目我們總結一下楊氏矩陣的兩個特點：

1. 同一行的元素由左向右依次遞增
2. 同一列的元素從上到下依次遞增

通過這兩點我們會發現這個矩陣有兩個元素是特殊元素。

• 右上角元素3為其所在行最大的元素，為其所在列最小的元素
• 左下角元素7為其所在行最小的元素，為其所在列最大的元素

因此我們可以採用以下方法：
先拿出右上角的元素3來和所查詢的元素比較，如果3比要查詢的元素大，那說明該元素絕不可能在第一行，因此我們就可以直接排除一行的元素。如果3比要查詢的元素小，那說明該元素絕不可能在最後一列，因此我們就可以直接排除一列的元素

現在假設我們排除了一行的元素，那接下來的矩陣就變成了這樣：

這時6又變成了右上角的元素，然後重複上一步的操作，假設我們這次排除了一列的元素，那接下來的矩陣就變成了這樣：

當然上述舉例我是以右上角元素為基準的，如果以左下角元素為基準也可以得到相同的結果，大家不妨自己來試一下。

實現程式碼如下：

#include <stdio.h>

int find_num(int arr[3][3], int row, int col, int k)
{
	int x = 0;
	int y = col-1;
	while (x<row && y>=0)
	{
		if (arr[x][y] == k)
		{
			printf
 
("下標為： %d %d\n", x, y);
			return 1;
		}
		else if (arr[x][y] > k)
			y--;
		else if (arr[x][y] < k)
			x++;
	}
	return 0;
}

int main()
{
	int arr[3][3] = { 1, 2, 3, 4, 5, 6, 7, 8, 9 };
	int ret = find_num(arr, 3, 3, 7);
	if (ret == 1)
		printf("找到了\n");
	else
		printf("找不到\n");
	return 0;
}

這裡我們將查詢楊氏矩陣元素的過程封裝在一個函式中。函式接收4個引數，分別是二維陣列的地址，行數，列數和要查詢的元素。通過返回值來判斷是否找到。

在函式內部定義一個座標（x, y）表示右上角元素，當x等於行數是說明已經越界(陣列下標是從0開始的)，那要查詢的元素必然不存在。當列數小於0也一樣。

當排除一行的時候，給x的值加1即可；排除一列的時候，給y的值減1即可。

執行結果：