移動零
五星
LeetCode:移動零
題目描述:
給定一個數組 nums,編寫一個函數將所有 0 移動到數組的末尾,同時保持非零元素的相對順序。
示例:
輸入: [0,1,0,3,12]
輸出: [1,3,12,0,0]
思想:
直接在正確的位置賦值,覆蓋掉的數不影響最終結果,最後騰出的幾位都賦值0
代碼:
class Solution {
public void moveZeroes(int[] nums) {
int i = 0;
for(int num : nums){
if(num!=0) nums[i++] = num;
}
for(;i<nums.length;++i){
nums[i] = 0;
}
}
}
重塑矩陣
LeetCode:重塑矩陣
題目描述:
在MATLAB中,有一個非常有用的函數 reshape,它可以將一個矩陣重塑爲另一個大小不同的新矩陣,但保留其原始數據。
給出一個由二維數組表示的矩陣,以及兩個正整數r和c,分別表示想要的重構的矩陣的行數和列數。
重構後的矩陣需要將原始矩陣的所有元素以相同的行遍歷順序填充。
如果具有給定參數的reshape操作是可行且合理的,則輸出新的重塑矩陣;否則,輸出原始矩陣。
示例:
輸入:
nums =
[[1,2],
[3,4]]
r = 1, c = 4
輸出:
[[1,2,3,4]]
解釋:
行遍歷nums的結果是 [1,2,3,4]。新的矩陣是 1 * 4 矩陣, 用之前的元素值一行一行填充新矩陣。
思想:
這題沒啥精妙的思想,就是有點複雜。
注意:index可以累加,不用每次都計算
代碼:
class Solution {
public int[][] matrixReshape(int[][] nums, int r, int c) {
int m = nums.length;
int n = nums[0].length;
if(r*c!=m*n) return nums;
int[][] res = new int[r][c];
int index=0;
for(int i=0;i<r;++i){
for(int j=0;j<c;++j){
res[i][j] = nums[index/n][index%n];
index++;
}
}
return res;
}
}