Question

1460. 通过翻转子数组使两个数组相等

English Version

题目描述

给你两个长度相同的整数数组 target 和 arr 。每一步中，你可以选择 arr 的任意 非空子数组 并将它翻转。你可以执行此过程任意次。

_如果你能让 arr 变得与 target 相同，返回 True；否则，返回 False 。_

 

示例 1：

输入：target = [1,2,3,4], arr = [2,4,1,3]
输出：true
解释：你可以按照如下步骤使 arr 变成 target：
1- 翻转子数组 [2,4,1] ，arr 变成 [1,4,2,3]
2- 翻转子数组 [4,2] ，arr 变成 [1,2,4,3]
3- 翻转子数组 [4,3] ，arr 变成 [1,2,3,4]
上述方法并不是唯一的，还存在多种将 arr 变成 target 的方法。

示例 2：

输入：target = [7], arr = [7]
输出：true
解释：arr 不需要做任何翻转已经与 target 相等。

示例 3：

输入：target = [3,7,9], arr = [3,7,11]
输出：false
解释：arr 没有数字 9 ，所以无论如何也无法变成 target 。

 

提示：

• target.length == arr.length
• 1 <= target.length <= 1000
• 1 <= target[i] <= 1000
• 1 <= arr[i] <= 1000

解法

方法一：排序

分别对数组 $arr$ 和 $target$ 排序，然后比较两数组对应位置的元素是否相等。相等则满足条件。

时间复杂度 $O(nlogn)$，空间复杂度 $O(logn)$。其中 $n$ 是数组 $arr$ 的长度，快排的平均递归深度为 $O(logn)$。

class Solution:
  def canBeEqual(self, target: List[int], arr: List[int]) -> bool:
    target.sort()
    arr.sort()
    return target == arr

class Solution {
  public boolean canBeEqual(int[] target, int[] arr) {
    Arrays.sort(target);
    Arrays.sort(arr);
    return Arrays.equals(target, arr);
  }
}

class Solution {
public:
  bool canBeEqual(vector<int>& target, vector<int>& arr) {
    sort(target.begin(), target.end());
    sort(arr.begin(), arr.end());
    return target == arr;
  }
};

func canBeEqual(target []int, arr []int) bool {
  sort.Ints(target)
  sort.Ints(arr)
  for i, v := range target {
    if v != arr[i] {
      return false
    }
  }
  return true
}

function canBeEqual(target: number[], arr: number[]): boolean {
  target.sort((a, b) => a - b);
  arr.sort((a, b) => a - b);
  const n = arr.length;
  for (let i = 0; i < n; i++) {
    if (target[i] !== arr[i]) {
      return false;
    }
  }
  return true;
}

impl Solution {
  pub fn can_be_equal(mut target: Vec<i32>, mut arr: Vec<i32>) -> bool {
    target.sort();
    arr.sort();
    target == arr
  }
}

class Solution {
  /**
   * @param Integer[] $target
   * @param Integer[] $arr
   * @return Boolean
   */
  function canBeEqual($target, $arr) {
    sort($target);
    sort($arr);
    return $target === $arr;
  }
}

bool canBeEqual(int* target, int targetSize, int* arr, int arrSize) {
  int count[1001] = {0};
  for (int i = 0; i < targetSize; i++) {
    count[target[i]]++;
    count[arr[i]]--;
  }
  for (int i = 0; i < 1001; i++) {
    if (count[i] != 0) {
      return false;
    }
  }
  return true;
}

方法二：数组/哈希表

由于两数组的数据范围都是 $1 \leq x \leq 1000$，因此我们可以使用数组或哈希表来记录每个数字出现的次数。

时间复杂度 $O(n)$，空间复杂度 $O(C)$。其中 $n$ 是数组 $arr$ 的长度，而 $C$ 是数组 $arr$ 元素的值域大小。

class Solution:
  def canBeEqual(self, target: List[int], arr: List[int]) -> bool:
    return Counter(target) == Counter(arr)

class Solution {
  public boolean canBeEqual(int[] target, int[] arr) {
    int[] cnt1 = new int[1001];
    int[] cnt2 = new int[1001];
    for (int v : target) {
      ++cnt1[v];
    }
    for (int v : arr) {
      ++cnt2[v];
    }
    return Arrays.equals(cnt1, cnt2);
  }
}

class Solution {
public:
  bool canBeEqual(vector<int>& target, vector<int>& arr) {
    vector<int> cnt1(1001);
    vector<int> cnt2(1001);
    for (int& v : target) ++cnt1[v];
    for (int& v : arr) ++cnt2[v];
    return cnt1 == cnt2;
  }
};

func canBeEqual(target []int, arr []int) bool {
  cnt1 := make([]int, 1001)
  cnt2 := make([]int, 1001)
  for _, v := range target {
    cnt1[v]++
  }
  for _, v := range arr {
    cnt2[v]++
  }
  for i, v := range cnt1 {
    if v != cnt2[i] {
      return false
    }
  }
  return true
}

function canBeEqual(target: number[], arr: number[]): boolean {
  const n = target.length;
  const count = new Array(1001).fill(0);
  for (let i = 0; i < n; i++) {
    count[target[i]]++;
    count[arr[i]]--;
  }
  return count.every(v => v === 0);
}

impl Solution {
  pub fn can_be_equal(mut target: Vec<i32>, mut arr: Vec<i32>) -> bool {
    let n = target.len();
    let mut count = [0; 1001];
    for i in 0..n {
      count[target[i] as usize] += 1;
      count[arr[i] as usize] -= 1;
    }
    count.iter().all(|v| *v == 0)
  }
}

方法三

class Solution:
  def canBeEqual(self, target: List[int], arr: List[int]) -> bool:
    cnt = [0] * 1001
    for a, b in zip(target, arr):
      cnt[a] += 1
      cnt[b] -= 1
    return all(v == 0 for v in cnt)

class Solution {
  public boolean canBeEqual(int[] target, int[] arr) {
    int[] cnt = new int[1001];
    for (int v : target) {
      ++cnt[v];
    }
    for (int v : arr) {
      if (--cnt[v] < 0) {
        return false;
      }
    }
    return true;
  }
}

class Solution {
public:
  bool canBeEqual(vector<int>& target, vector<int>& arr) {
    vector<int> cnt(1001);
    for (int& v : target) ++cnt[v];
    for (int& v : arr)
      if (--cnt[v] < 0) return false;
    return true;
  }
};

func canBeEqual(target []int, arr []int) bool {
  cnt := make([]int, 1001)
  for _, v := range target {
    cnt[v]++
  }
  for _, v := range arr {
    cnt[v]--
    if cnt[v] < 0 {
      return false
    }
  }
  return true
}