給你一棵 n 個節(jié)點的樹，編號從 0 到 n - 1 ，以父節(jié)點數(shù)組 parent 的形式給出，其中 parent[i] 是第 i 個節(jié)點的父節(jié)點。樹的根節(jié)點為 0 號節(jié)點，所以 parent[0] = -1 ，因為它沒有父節(jié)點。你想要設(shè)計一個數(shù)據(jù)結(jié)構(gòu)實現(xiàn)樹里面對節(jié)點的加鎖，解鎖和升級操作。

數(shù)據(jù)結(jié)構(gòu)需要支持如下函數(shù)：

• Lock：指定用戶給指定節(jié)點 上鎖 ，上鎖后其他用戶將無法給同一節(jié)點上鎖。只有當(dāng)節(jié)點處于未上鎖的狀態(tài)下，才能進(jìn)行上鎖操作。
• Unlock：指定用戶給指定節(jié)點 解鎖 ，只有當(dāng)指定節(jié)點當(dāng)前正被指定用戶鎖住時，才能執(zhí)行該解鎖操作。
• Upgrade：指定用戶給指定節(jié)點 上鎖 ，并且將該節(jié)點的所有子孫節(jié)點 解鎖 。只有如下 3 個條件 全部 滿足時才能執(zhí)行升級操作：

指定節(jié)點當(dāng)前狀態(tài)為未上鎖。

指定節(jié)點至少有一個上鎖狀態(tài)的子孫節(jié)點（可以是 任意 用戶上鎖的）。

指定節(jié)點沒有任何上鎖的祖先節(jié)點。
請你實現(xiàn) LockingTree 類：

LockingTree(int[] parent) 用父節(jié)點數(shù)組初始化數(shù)據(jù)結(jié)構(gòu)。 lock(int num, int user) 如果 id 為 user 的用戶可以給節(jié)點 num 上鎖，那么返回 true ，否則返回 false 。如果可以執(zhí)行此操作，節(jié)點 num 會被 id 為 user 的用戶 上鎖 。 unlock(int num, int user) 如果 id 為 user 的用戶可以給節(jié)點 num 解鎖，那么返回 true ，否則返回 false 。如果可以執(zhí)行此操作，節(jié)點 num 變?yōu)?未上鎖 狀態(tài)。 upgrade(int num, int user) 如果 id 為 user 的用戶可以給節(jié)點 num 升級，那么返回 true ，否則返回 false 。如果可以執(zhí)行此操作，節(jié)點 num 會被 升級 。

示例 1：

輸入： ["LockingTree", "lock", "unlock", "unlock", "lock", "upgrade", "lock"] [[[-1, 0, 0, 1, 1, 2, 2]], [2, 2], [2, 3], [2, 2], [4, 5], [0, 1], [0, 1]] 輸出： [null, true, false, true, true, true, false]解釋： LockingTree lockingTree = new LockingTree([-1, 0, 0, 1, 1, 2, 2]); lockingTree.lock(2, 2); // 返回 true ，因為節(jié)點 2 未上鎖。// 節(jié)點 2 被用戶 2 上鎖。 lockingTree.unlock(2, 3); // 返回 false ，因為用戶 3 無法解鎖被用戶 2 上鎖的節(jié)點。 lockingTree.unlock(2, 2); // 返回 true ，因為節(jié)點 2 之前被用戶 2 上鎖。// 節(jié)點 2 現(xiàn)在變?yōu)槲瓷湘i狀態(tài)。 lockingTree.lock(4, 5); // 返回 true ，因為節(jié)點 4 未上鎖。// 節(jié)點 4 被用戶 5 上鎖。 lockingTree.upgrade(0, 1); // 返回 true ，因為節(jié)點 0 未上鎖且至少有一個被上鎖的子孫節(jié)點（節(jié)點 4）。// 節(jié)點 0 被用戶 1 上鎖，節(jié)點 4 變?yōu)槲瓷湘i。 lockingTree.lock(0, 1); // 返回 false ，因為節(jié)點 0 已經(jīng)被上鎖了。

提示：

• n == parent.length
• 2 <= n <= 2000
• 對于 i != 0 ，滿足 0 <= parent[i] <= n - 1
• parent[0] == -1
• 0 <= num <= n - 1
• 1 <= user <= 104
• parent 表示一棵合法的樹。
• lock ，unlock 和 upgrade 的調(diào)用 總共 不超過 2000 次。

解題思路

使用map維護(hù)被鎖節(jié)點和加鎖者的關(guān)系，再使用一個map維護(hù)父節(jié)點和對應(yīng)子節(jié)點之間的關(guān)系

lock

判斷是否有鎖，如果沒有，則可以直接加鎖

unlock

判斷當(dāng)前鎖的加鎖者是否為自己

upgrade

先向上遍歷祖先節(jié)點和加鎖節(jié)點，判斷是否有加鎖

再遍歷子孫節(jié)點，判斷是否加鎖

給所有子孫節(jié)點解鎖，給當(dāng)前節(jié)點加鎖

代碼

class LockingTree {int[] p;Map<Integer, Integer> locked = new HashMap<>();Map<Integer,Set<Integer>> child=new HashMap<>();public LockingTree(int[] parent) {p = parent;for (int i = 0; i < parent.length; i++) {if (!child.containsKey(p[i]))child.put(p[i],new HashSet<>());child.get(p[i]).add(i);}}public boolean lock(int num, int user) {if (locked.containsKey(num) )return false;locked.put(num, user);return true;}public boolean unlock(int num, int user) {if (locked.containsKey(num) && locked.get(num) == user) {locked.remove(num);return true;} else return false;}public boolean dfs(int num) {while (num!=-1){if(locked.containsKey(num))return false;num=p[num];}return true;}public boolean ddfs(int num,int f) {if (locked.containsKey(num)&&num!=f){return true;}boolean res=false;if (child.containsKey(num)){for (Integer integer : child.get(num)) {res|=ddfs(integer,f);}}return res;}public void dddfs(int num) {locked.remove(num);if (child.containsKey(num)){for (Integer integer : child.get(num)) {dddfs(integer);}}}public boolean upgrade(int num, int user) {if (dfs(num)&&ddfs(num,num)){dddfs(num);locked.put(num, user);return true;}return false;}}/*** Your LockingTree object will be instantiated and called as such:* LockingTree obj = new LockingTree(parent);* boolean param_1 = obj.lock(num,user);* boolean param_2 = obj.unlock(num,user);* boolean param_3 = obj.upgrade(num,user);*//*** Your LockingTree object will be instantiated and called as such:* LockingTree obj = new LockingTree(parent);* boolean param_1 = obj.lock(num,user);* boolean param_2 = obj.unlock(num,user);* boolean param_3 = obj.upgrade(num,user);*/

總結(jié)

以上是生活随笔為你收集整理的5848. 树上的操作的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。