目标
给你一个下标从 0 开始的整数数组 nums ,表示一些石块的初始位置。再给你两个长度 相等 下标从 0 开始的整数数组 moveFrom 和 moveTo 。
在 moveFrom.length 次操作内,你可以改变石块的位置。在第 i 次操作中,你将位置在 moveFrom[i] 的所有石块移到位置 moveTo[i] 。
完成这些操作后,请你按升序返回所有 有 石块的位置。
注意:
- 如果一个位置至少有一个石块,我们称这个位置 有 石块。
- 一个位置可能会有多个石块。
示例 1:
输入:nums = [1,6,7,8], moveFrom = [1,7,2], moveTo = [2,9,5]
输出:[5,6,8,9]
解释:一开始,石块在位置 1,6,7,8 。
第 i = 0 步操作中,我们将位置 1 处的石块移到位置 2 处,位置 2,6,7,8 有石块。
第 i = 1 步操作中,我们将位置 7 处的石块移到位置 9 处,位置 2,6,8,9 有石块。
第 i = 2 步操作中,我们将位置 2 处的石块移到位置 5 处,位置 5,6,8,9 有石块。
最后,至少有一个石块的位置为 [5,6,8,9] 。
示例 2:
输入:nums = [1,1,3,3], moveFrom = [1,3], moveTo = [2,2]
输出:[2]
解释:一开始,石块在位置 [1,1,3,3] 。
第 i = 0 步操作中,我们将位置 1 处的石块移到位置 2 处,有石块的位置为 [2,2,3,3] 。
第 i = 1 步操作中,我们将位置 3 处的石块移到位置 2 处,有石块的位置为 [2,2,2,2] 。
由于 2 是唯一有石块的位置,我们返回 [2] 。
说明:
- 1 <= nums.length <= 10^5
- 1 <= moveFrom.length <= 10^5
- moveFrom.length == moveTo.length
- 1 <= nums[i], moveFrom[i], moveTo[i] <= 10^9
- 测试数据保证在进行第 i 步操作时,moveFrom[i] 处至少有一个石块。
思路
有一个数组 nums
,其元素值表示该位置上有石头,另用两个长度相同的数组 moveFrom
moveTo
,表示将from位置上 所有 的石头搬到to的操作。问依次完成这些操作后,哪些位置上有石头,按位置从小到大顺序返回。
很容易想到用哈希表记录位置上 石头的数量,然后根据操作增减相应位置的石头。最后遍历找出石头数量大于0的位置,然后排序即可。每次移走所有石头,所有不用计数。
换一个角度考虑,to位置上是否一定有石头?不一定,因为可能被后序操作移走。这样我们就不能整体考虑,通过集合操作得到结果。
再换一个角度考虑,能否根据from to 建图,我们只需求出出度为0的节点即可。节点数量太多还是算了。
还是常规做法吧。
看了题解,大家都是这么做的。无非是有的用 HashSet 有的用 HashMap。最快的做法是最后排序的时候转为数组,使用数组排序
代码
/**
* @date 2024-07-24 21:43
*/
public class RelocateMarbles2766 {
public List<Integer> relocateMarbles(int[] nums, int[] moveFrom, int[] moveTo) {
Set<Integer> hasStone = new HashSet<>(nums.length);
for (int num : nums) {
hasStone.add(num);
}
int n = moveFrom.length;
for (int i = 0; i < n; i++) {
hasStone.remove(moveFrom[i]);
hasStone.add(moveTo[i]);
}
List<Integer> res = new ArrayList<>(hasStone);
Collections.sort(res);
return res;
}
}