1. 程式人生 > 實用技巧 >11.14

11.14

1122. 陣列的相對排序

難度簡單119

給你兩個陣列,arr1arr2

  • arr2 中的元素各不相同
  • arr2 中的每個元素都出現在 arr1

arr1 中的元素進行排序,使 arr1 中項的相對順序和 arr2 中的相對順序相同。未在 arr2 中出現過的元素需要按照升序放在 arr1 的末尾。

示例:

輸入:arr1 = [2,3,1,3,2,4,6,7,9,2,19], arr2 = [2,1,4,3,9,6]
輸出:[2,2,2,1,4,3,3,9,6,7,19]

提示:

  • arr1.length, arr2.length <= 1000
  • 0 <= arr1[i], arr2[i] <= 1000
  • arr2 中的元素 arr2[i] 各不相同
  • arr2 中的每個元素 arr2[i] 都出現在 arr1

解答:直接將arr2存入hash表,然後根據value值對arr1進行排序

class Solution {
public:
    vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2) {
        unordered_map<int, int>cnt;
        for(int i = 0; i < arr2.size(); i++)cnt[arr2[i]] = arr2.size() - i;
        sort(arr1.begin(), arr1.end(), [&](int a, int b){
            if(cnt[a] != cnt[b])return cnt[a] > cnt[b];
            return a < b; 
        });
        return arr1;
    }
};

100. 相同的樹

難度簡單509

給定兩個二叉樹,編寫一個函式來檢驗它們是否相同。

如果兩個樹在結構上相同,並且節點具有相同的值,則認為它們是相同的。

示例 1:

輸入:       1         1
          / \       / \
         2   3     2   3

        [1,2,3],   [1,2,3]

輸出: true

示例 2:

輸入:      1          1
          /           \
         2             2

        [1,2],     [1,null,2]

輸出: false

示例 3:

輸入:       1         1
          / \       / \
         2   1     1   2

        [1,2,1],   [1,1,2]

輸出: false

解答:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    bool isSameTree(TreeNode* p, TreeNode* q) {
        if(!p && !q)return true;
         if(p == NULL|| q == NULL || p->val != q->val)return false;
         
        return isSameTree(p->right, q->right) && isSameTree(p->left, q->left);
    }
};

112. 路徑總和

難度簡單462

給定一個二叉樹和一個目標和,判斷該樹中是否存在根節點到葉子節點的路徑,這條路徑上所有節點值相加等於目標和。

說明: 葉子節點是指沒有子節點的節點。

示例:
給定如下二叉樹,以及目標和 sum = 22

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \      \
        7    2      1

返回 true, 因為存在目標和為 22 的根節點到葉子節點的路徑 5->4->11->2

解答:簡單的dfs

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool hasPathSum(TreeNode* root, int sum) {
        if(!root)return false;
        sum = sum - root->val;
        if(!root->right && !root->left)return !sum;
        return root->left && hasPathSum(root->left, sum) ||  root->right && hasPathSum(root->right, sum);
    }
};

113. 路徑總和 II

難度中等383

給定一個二叉樹和一個目標和,找到所有從根節點到葉子節點路徑總和等於給定目標和的路徑。

說明: 葉子節點是指沒有子節點的節點。

示例:
給定如下二叉樹,以及目標和 sum = 22

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

返回:

[
   [5,4,11,2],
   [5,8,4,5]
]

解答:dfs回溯

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>>ans;
    vector<int>path;
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        if(root)dfs(root, sum);
        return ans;
    }
    void dfs(TreeNode* root, int sum){
        path.push_back(root->val);
        sum -= root->val;
        if(!root->right && !root->left){
            if(!sum)ans.push_back(path);
        }else{
            if(root->right)dfs(root->right, sum);
            if(root->left)dfs(root->left, sum);
        }
        path.pop_back();
    }
};