從矩陣中取出子矩陣

探索一個簡單的算法,從矩陣中取出一個小矩陣應該怎樣實現。

具體什麼意思呢,比方說我們現在有一個int型的數組,將其用矩陣的形式寫出來就會長這個樣子:

int[] arr = {1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25};

用類似矩陣的形狀寫出來如下:

int[] arr = {
 1,  2,  3,  4,  5, 
 6,  7,  8,  9,  10,
11, 12, 13, 14, 15, 
16, 17, 18, 19, 20, 
21, 22, 23, 24, 25};

以上是一個5X5的矩陣形狀,我們如果要在這裏面取一個子矩陣,要求就是必須子矩陣的寬高都是小於等於父矩陣,在這裏就是小於等於5。

假設我們要在這個矩陣裏面取出一個子矩陣,所需要的條件是:
父矩陣數據,父矩陣寬高,子矩陣寬高,子矩陣左上角的XY座標
所以我們需要7個條件才能準確的知道子矩陣的信息

假設我們需要一個長爲2,寬爲2,位置在X=2,Y=3的位置(0是起始位置),那麼我們就應該得到如下子矩陣:
int [] child = {
18,19,
23,24
}

現在我們需要製作一個黑盒子,輸入7位元素,黑盒子給我一個我需要的子矩陣。

直接上黑盒子的原理好了

    /**
     * @param arr 待截取數組
     * @param x   待截取位置待X座標
     * @param y   待截取位置的Y座標
     * @param rw  需要截取的長度
     * @param rh  需要截取的寬度
     * @param w   arr數組的原始寬度
     * @param h   arr數組的原始高度
     * @return 已截取完畢的數組
     */
    private static int[] getSpecilArr(int[] arr, int x, int y, int rw, int rh, int w, int h) {

        if (rw > w || rh > h || x > w || y > h || x + rw > w || y + rh > h) {
            throw new IllegalStateException();
        }

        int[] result = new int[rw * rh];

        for (int i = 0; i < rh; i++) {
            for (int j = 0; j < rw; j++) {
                result[i * rw + j] = arr[(i + y) * w + x + j];
            }
        }

        return result;

    }

結束!
(觀衆:怎麼那麼突然!!!)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章