1. 程式人生 > 其它 >[CUDA]CUDA程式設計實戰一——瞭解CUDA及獲取GPU資訊

[CUDA]CUDA程式設計實戰一——瞭解CUDA及獲取GPU資訊

在一個 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; } }