-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add more comments and refactoring.
- Loading branch information
Tzu Hsiang Lin
committed
Nov 28, 2019
1 parent
eae0d1c
commit 8204997
Showing
6 changed files
with
73 additions
and
14 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,48 @@ | ||
''' | ||
1122. Relative Sort Array | ||
https://leetcode.com/problems/relative-sort-array/ | ||
Given two arrays arr1 and arr2, | ||
the elements of arr2 are distinct, | ||
and all elements in arr2 are also in arr1. | ||
Sort the elements of arr1 such that | ||
the relative ordering of items in arr1 are the same as in arr2. | ||
Elements that don't appear in arr2 should be | ||
placed at the end of arr1 in ascending order. | ||
Example 1: | ||
Input: arr1 = [2,3,1,3,2,4,6,7,9,2,19], arr2 = [2,1,4,3,9,6] | ||
Output: [2,2,2,1,4,3,3,9,6,7,19] | ||
Constraints: | ||
arr1.length, arr2.length <= 1000 | ||
0 <= arr1[i], arr2[i] <= 1000 | ||
Each arr2[i] is distinct. | ||
Each arr2[i] is in arr1. | ||
''' | ||
from typing import * | ||
import collections | ||
|
||
class Solution: | ||
def relativeSortArray(self, arr1: List[int], arr2: List[int]) -> List[int]: | ||
ans = [] | ||
c1 = collections.Counter(arr1) | ||
for num in arr2: | ||
ans.extend([num] * c1[num]) | ||
del c1[num] | ||
remain = list(c1.keys()) | ||
for num in sorted(remain): | ||
# remain num should appear the same times | ||
ans.extend([num] * c1[num]) | ||
return ans | ||
|
||
arr1 = [2,3,1,3,2,4,6,7,9,2,19] | ||
arr2 = [2,1,4,3,9,6] | ||
print(Solution().relativeSortArray(arr1, arr2)) | ||
# [2,2,2,1,4,3,3,9,6,7,19] | ||
|
||
arr1 = [2,21,43,38,0,42,33,7,24,13,12,27,12,24,5,23,29,48,30,31] | ||
arr2 = [2,42,38,0,43,21] | ||
print(Solution().relativeSortArray(arr1, arr2)) | ||
# [2,42,38,0,43,21,5,7,12,12,13,23,24,24,27,29,30,31,33,48] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters