[CUDA]CUDA程式設計實戰一——瞭解CUDA及獲取GPU資訊
阿新 • • 發佈:2021-06-11
在一個 n * m 的二維陣列中,每一行都按照從左到右遞增的順序排序,每一列都按照從上到下遞增的順序排序。
請完成一個高效的函式,輸入這樣的一個二維陣列和一個整數,判斷陣列中是否含有該整數。
示例:
現有矩陣 matrix 如下:
[
[1, 4, 7, 11, 15],
[2, 5, 8, 12, 19],
[3, 6, 9, 16, 22],
[10, 13, 14, 17, 24],
[18, 21, 23, 26, 30]
]
給定 target=5
,返回true
。
給定target=20
,返回false
。
限制:
0 <= n <= 1000
0 <= m <= 1000
題解:
由題意,每一行按從左到右遞增,每一列按從上到下遞增。
我們選取右上角的點為起始點,開始搜尋。
當該點大於target,則向左移動(向小的方向移動),
當該點小於target,則向下移動(向大的方向移動)。
public class offer04 { public static void main(String[] args) { // TODO Auto-generated method stub int[][] matrix = new int[][]{ {1, 4, 7, 11, 15}, {2, 5, 8, 12, 19}, {3, 6, 9, 16, 22}, {10, 13, 14, 17, 24}, {18, 21, 23, 26, 30} }; int target = 20; System.out.println(findNumberIn2DArray(matrix,target)); }public static boolean findNumberIn2DArray(int[][] matrix, int target) { if(matrix==null||matrix.length==0||matrix[0].length==0){ return false; } int rows = matrix.length; int cols = matrix[0].length; int i = 0,j = cols-1; while(matrix[i][j]!=target){ if(matrix[i][j]>target){ j = j-1; }else if(matrix[i][j]<target){ i = i+1; } if(i>=rows||j<0){//當i或者j超出邊界時,說明沒有找到,返回false return false; } //System.out.println("i:"+i+",j:"+j+",matrix:"+matrix[i][j]); } return true; } }