|
| 1 | +/* |
| 2 | +runtime 8 ms, beats 4.87% |
| 3 | +memory 55.44 MB, beats 54.17% |
| 4 | +
|
| 5 | +time complexity: O(n) |
| 6 | +- nums 각각의 원소를 소인수분해 : O(10 * n) = O(n) |
| 7 | +- ans[] 배열 계산 : O(10 * n) |
| 8 | + - i, j, k의 3중 for-loop가 O(n * 10 * 4) 처럼 보일 수 있는데요(10은 소인수 분해 base의 개수, 4는 power중 최대값), |
| 9 | + - 실제로는 k가 큰 경우에는 j-loop가 일찍 끝나고 (예컨대 nums[i] = 2^4 인 경우에는, base가 2에서 끝나니 j-loop가 j=1에서 끝남) |
| 10 | + - j가 큰 경우에는 k-loop가 일찍 끝나므로 (예컨대 num[i] = 29인 경우에는, k-loop가 1에서 끝남) |
| 11 | + - 최악의 경우에 O(n * 10) 정도로 보는 게 타당하다고 생각합니다. |
| 12 | +
|
| 13 | +space complexity: O(1) |
| 14 | +
|
| 15 | +[풀이] |
| 16 | +나눗셈이 금지되어 있기 때문에, 최대한 곱할 수 있는 만큼을 미리 곱해놓고, i번째 원소마다 부족한 부분만큼을 나중에 채워 넣어 곱해주는 방식으로 접근했습니다. |
| 17 | +nums[i]의 절대값이 30 이하인 점에 착안해서, nums[i]는 많아야 10개의 base로 소인수 분해가 가능하다는 점을 떠올렸습니다. 따라서 각각의 base마다 (nums 전체에서 등장한 base의 지수의 총합) - (nums[i]의 base의 지수 중 가장 큰 값) 만큼의 지수로 미리 곱해주고, i에 대한 iteration에서 부족한 지수만큼을 보충해서 곱해주었습니다. |
| 18 | +(풀이를 쓰다 보니, 실제 코드에서 이렇게 미리 곱해 놓은 수를 base라고 명명한 게 혼동될 수 있겠네요...^^;) |
| 19 | +
|
| 20 | +이 풀이에서 아쉬운 점이 여전히 있습니다. 시간 복잡도를 따지면 O(10*n) = O(n)이니 문제의 조건을 맞췄다고 할 수도 있겠지만, 상수가 좀 크다는 점이 마음에 걸리네요. |
| 21 | +(for (int j = 0; j < 10 && abs > 1; j++) <--- 여기에서 abs == 1이 되면 for-loop를 빠져나가게 했지만, nums[i]가 모두 29로 차있는 edge case라면 10*n을 꽉 채우게 되니까요.) |
| 22 | +그래도 어쨌든 문제의 조건을 최대한 활용해서 시도해볼만한 접근이라고 생각합니다. |
| 23 | +
|
| 24 | +이 풀이 말고도, nums[]를 2개로 나눈 블럭, 4개로 나눈 블럭, 8개로 나눈 블럭, ... 이런 식으로 사이즈 별로 블럭을 나눠두고, 각각의 블럭 내부의 곱을 미리 계산해두는 방식도 생각해보았습니다. 이러면 시간 복잡도와 공간 복잡도가 모두 O(nlogn)이 나올 것 같습니다. (사실 이 풀이를 가장 처음에 떠올렸습니다만, O(n)으로 줄이는 고민을 하다가 현재 제출한 풀이를 떠올리고서는 이 풀이는 구현을 안 했습니다. 시간이 되면 이렇게도 풀어보고, 좀 더 디벨롭을 해봐야겠네요.) |
| 25 | +*/ |
| 26 | +class Solution { |
| 27 | + public int[] productExceptSelf(int[] nums) { |
| 28 | + int n = nums.length; |
| 29 | + |
| 30 | + int[] primes = {2, 3, 5, 7, 11, 13, 17, 19, 23, 29}; |
| 31 | + int[] sumPowers = new int[10]; |
| 32 | + int[] maxPowers = new int[10]; |
| 33 | + |
| 34 | + int base = 1; |
| 35 | + |
| 36 | + int[] ans = new int[n]; |
| 37 | + |
| 38 | + for (int i = 0; i < n; i++) { |
| 39 | + if (nums[i] == 0) { |
| 40 | + int x = 1; |
| 41 | + for (int j = 0; j < n; j++) { |
| 42 | + if (j == i) { |
| 43 | + continue; |
| 44 | + } |
| 45 | + ans[j] = 0; |
| 46 | + x *= nums[j]; |
| 47 | + } |
| 48 | + ans[i] = x; |
| 49 | + |
| 50 | + return ans; |
| 51 | + } |
| 52 | + |
| 53 | + int abs; |
| 54 | + if (nums[i] > 0) { |
| 55 | + abs = nums[i]; |
| 56 | + } else { |
| 57 | + abs = -nums[i]; |
| 58 | + base = -base; |
| 59 | + } |
| 60 | + |
| 61 | + for (int j = 0; j < 10 && abs > 1; j++) { |
| 62 | + int curPower = 0; |
| 63 | + while (abs % primes[j] == 0) { |
| 64 | + sumPowers[j]++; |
| 65 | + curPower++; |
| 66 | + abs /= primes[j]; |
| 67 | + } |
| 68 | + if (curPower > maxPowers[j]) { |
| 69 | + maxPowers[j] = curPower; |
| 70 | + } |
| 71 | + } |
| 72 | + } |
| 73 | + |
| 74 | + for (int i = 0; i < 10; i++) { |
| 75 | + for (int j = 0; j < sumPowers[i] - maxPowers[i]; j++) { |
| 76 | + base *= primes[i]; |
| 77 | + } |
| 78 | + } |
| 79 | + |
| 80 | + for (int i = 0; i < n; i++) { |
| 81 | + ans[i] = base; |
| 82 | + for (int j = 0; j < 10; j++) { |
| 83 | + int tmp = nums[i]; |
| 84 | + int power = 0; |
| 85 | + while (tmp % primes[j] == 0) { |
| 86 | + power++; |
| 87 | + tmp /= primes[j]; |
| 88 | + } |
| 89 | + for (int k = 0; k < maxPowers[j] - power; k++) { |
| 90 | + ans[i] *= primes[j]; |
| 91 | + } |
| 92 | + } |
| 93 | + if (nums[i] < 0) { |
| 94 | + ans[i] = -ans[i]; |
| 95 | + } |
| 96 | + } |
| 97 | + |
| 98 | + |
| 99 | + return ans; |
| 100 | + } |
| 101 | +} |
0 commit comments