|
| 1 | +// 문제 풀이 흐름 : |
| 2 | +// 전체의 합에서 자기 자신만 뺀것을 생각하듯이, 전체의 곱에서 자기 자신만 나눈 것을 생각하자 |
| 3 | +// 이때, 중요한 것은 integer overflow와 0이 여러번 나올 떄 예외처리 |
| 4 | +// integer overflow : 문제에서는 답이 32 bit integer 라고 했지, 그 과정에서 생길 수 있는 곱들이 int범위라고 단정지을 수 없음 |
| 5 | +// 0의 개수에 따라 다르게 계산 : |
| 6 | +// 0이 없을 때 : 상관없음. |
| 7 | +// 1개만 있을 때 : 0인 곳에만 product 기입 |
| 8 | +// 2개만 있을 때 : 모두 0이 된다. |
| 9 | + |
| 10 | +// n = nums.length 라 할때 |
| 11 | +// 시간복잡도 : O(n) |
| 12 | +// 공간복잡도 : O(n) |
| 13 | + |
| 14 | + |
| 15 | +class Solution { |
| 16 | + |
| 17 | + static { |
| 18 | + Solution warmup = new Solution(); |
| 19 | + for (int i = 0; i < 500; ++i) { |
| 20 | + warmup.productExceptSelf(new int[2]); |
| 21 | + } |
| 22 | + } |
| 23 | + |
| 24 | + |
| 25 | + public int[] productExceptSelf(int[] nums) { |
| 26 | + int[] answer = new int[nums.length]; |
| 27 | + |
| 28 | + Arrays.fill(answer, 1); |
| 29 | + |
| 30 | + int leftProduct = 1; |
| 31 | + for (int i = 0 ; i < nums.length - 1; i ++) { |
| 32 | + leftProduct *= nums[i]; |
| 33 | + answer[i + 1] *= leftProduct; |
| 34 | + } |
| 35 | + |
| 36 | + int rightProduct = 1; |
| 37 | + for (int i = nums.length - 1 ; i > 0 ; i --) { |
| 38 | + rightProduct *= nums[i]; |
| 39 | + answer[i - 1] *= rightProduct; |
| 40 | + } |
| 41 | + |
| 42 | + return answer; |
| 43 | + } |
| 44 | + |
| 45 | + public int[] prevSubmission(int[] nums) { |
| 46 | + int zeros = 0; |
| 47 | + long totalProduct = 1; |
| 48 | + |
| 49 | + for (int number : nums) { |
| 50 | + if (number == 0) { |
| 51 | + zeros ++; |
| 52 | + } |
| 53 | + totalProduct *= number; |
| 54 | + } |
| 55 | + |
| 56 | + int[] answer = new int[nums.length]; |
| 57 | + |
| 58 | + if (zeros == 0) { |
| 59 | + for (int i = 0 ; i < nums.length; i ++ ) { |
| 60 | + answer[i] = (int) (totalProduct / nums[i]); |
| 61 | + } |
| 62 | + } else if (zeros == 1) { |
| 63 | + int nonZeroProduct = 1; |
| 64 | + int zeroIdx = -1; |
| 65 | + for (int i = 0 ; i < nums.length; i ++ ) { |
| 66 | + if (nums[i] == 0) { |
| 67 | + zeroIdx = i; |
| 68 | + continue; |
| 69 | + } |
| 70 | + nonZeroProduct *= nums[i]; |
| 71 | + } |
| 72 | + answer[zeroIdx] = nonZeroProduct; |
| 73 | + } |
| 74 | + |
| 75 | + return answer; |
| 76 | + } |
| 77 | +} |
0 commit comments