22 * Copyright (C) 2018, Saul Lawliet <october dot sunbathe at gmail dot com>
33 * All rights reserved.
44 *
5+ * https://leetcode.com/problems/search-a-2d-matrix/
6+ * Q: 在一个有序的二维数组中, 判断是否存在一个特定的值
7+ *
58 * 转换一下矩阵下标, 同样是二分查找
69 */
710
811#include <stdbool.h>
912#include "c/data-structures/array.h"
1013#include "c/test.h"
1114
12- bool searchMatrix (int * * matrix , int matrixRowSize , int matrixColSize , int target ) {
13- int a = 0 , b = matrixRowSize * matrixColSize - 1 ;
15+ bool searchMatrix (int * * matrix , int matrixSize , int * matrixColSize , int target ) {
16+ int a = 0 , b = matrixSize * * matrixColSize - 1 ;
1417 while (a <= b ) {
1518 int i = (a + b ) / 2 ;
16- int v = matrix [i / matrixColSize ][i % matrixColSize ];
19+ int v = matrix [i / * matrixColSize ][i % * matrixColSize ];
1720 if (v > target ) {
1821 b = i - 1 ;
1922 } else if (v < target ) {
@@ -28,7 +31,8 @@ bool searchMatrix(int **matrix, int matrixRowSize, int matrixColSize, int target
2831void test (bool expect , const char * matrix , int target ) {
2932 arrayEntry * e = arrayParse2D (matrix , ARRAY_INT );
3033
31- EXPECT_EQ_INT (expect , searchMatrix (arrayValue (e ), arrayRow (e ), arrayCol (e ), target ));
34+ int col = arrayCol (e );
35+ EXPECT_EQ_INT (expect , searchMatrix (arrayValue (e ), arrayRow (e ), & col , target ));
3236
3337 arrayFree (e );
3438}
0 commit comments