Posts tagged as “math”

花花酱 LeetCode 7. Reverse Integer

By zxi on September 12, 2018

Problem

Given a 32-bit signed integer, reverse digits of an integer.

Example 1:

Input: 123
Output: 321

Example 2:

Input: -123
Output: -321

Example 3:

Input: 120
Output: 21

Note:
Assume we are dealing with an environment which could only store integers within the 32-bit signed integer range: [−2³¹, 2³¹− 1]. For the purpose of this problem, assume that your function returns 0 when the reversed integer overflows.

Solution: Simulation

Reverse digit by digit. Be careful about the overflow and negative numbers (especially in Python)

Time complexity: O(log(x)) ~ O(1)

Space complexity: O(log(x)) ~ O(1)

C++

// Author: Huahua

class Solution {

public:

int reverse(int x) {

int ans = 0;

while (x != 0) {

int r = x % 10;

if (ans > INT_MAX / 10 || ans < INT_MIN / 10) return 0;

ans = ans * 10 + r;

x /= 10;

}

return ans;

}

};

Java

class Solution {

public int reverse(int x) {

int ans = 0;

while (x != 0) {

int r = x % 10;

if (ans > Integer.MAX_VALUE / 10 || ans < Integer.MIN_VALUE / 10) return 0;

ans = ans * 10 + r;

x /= 10;

}

return ans;

}

Python3

class Solution:

def reverse(self, x):

min_val = -(2**31);

max_val = 2**31 - 1;

sign = -1 if x < 0 else 1

x = abs(x)

ans = 0

while x != 0:

ans = ans * 10 + (x % 10)

x //= 10

ans *= sign

if ans > max_val or ans < min_val: return 0

return ans

花花酱 LeetCode 633. Sum of Square Numbers

By zxi on August 22, 2018

Problem

Given a non-negative integer c, your task is to decide whether there’re two integers a and b such that a² + b² = c.

Example 1:

Input: 5
Output: True
Explanation: 1 * 1 + 2 * 2 = 5

Example 2:

Input: 3
Output: False

Solution: Math

Time complexity: O(sqrt(c))

Space complexity: O(1)

// Author: Huahua

// Running time: 4 ms

class Solution {

public:

bool judgeSquareSum(int c) {

int m = sqrt(c);

for (int a = 0; a <= m; ++a) {

int b = round(sqrt(c - a * a));

if (a * a + b * b == c) return true;

}

return false;

}

};

花花酱 LeetCode 891. Sum of Subsequence Widths

By zxi on August 20, 2018

Problem

Given an array of integers A, consider all non-empty subsequences of A.

For any sequence S, let the width of S be the difference between the maximum and minimum element of S.

Return the sum of the widths of all subsequences of A.

As the answer may be very large, return the answer modulo 10^9 + 7.

Example 1:

Input: [2,1,3]
Output: 6
Explanation:
Subsequences are [1], [2], [3], [2,1], [2,3], [1,3], [2,1,3].
The corresponding widths are 0, 0, 0, 1, 1, 2, 2.
The sum of these widths is 6.

Note:

1 <= A.length <= 20000
1 <= A[i] <= 20000

Solution: Math

Sort the array, for A[i]:

i numbers <= A[i]. A[i] is the upper bound of 2^i subsequences.
n – i – 1 numbers >= A[i]. A[i] is the lower bound of 2^(n – i – 1) subsequences.
A[i] contributes A[i] * 2^i – A[i] * 2^(n – i – 1) to the ans.

\(ans = \sum\limits_{i=0}^{n-1}A_{i}2^{i} – A_{i}2^{n – i – 1} =\sum\limits_{i=0}^{n-1}(A_i – A_{n-i-1})2^{i}\)

Time complexity: O(nlogn)

Space complexity: O(1)

// Author: Huahua

// Running time: 56 ms

class Solution {

public:

int sumSubseqWidths(vector<int>& A) {

constexpr long kMod = 1e9 + 7;

const int n = A.size();

sort(begin(A), end(A));

long ans = 0;

long p = 1;

for (int i = 0; i < n; ++i) {

ans = (ans + (A[i] - A[n - i - 1]) * p) % kMod;

p = (p << 1) % kMod;

}

return (ans + kMod) % kMod;

}

};

Time complexity: O(n)

Space complexity: O(n)

Counting sort

// Author: Huahua

// Running time: 16 / 44 ms (beats 100%)

static auto x = [](){std::ios::sync_with_stdio(false);cin.tie(nullptr);return nullptr;}();

class Solution {

public:

int sumSubseqWidths(vector<int>& A) {

constexpr long kMod = 1e9 + 7;

const int n = A.size();

countingSort(A);

long ans = 0;

long p = 1;

for (int i = 0; i < n; ++i) {

ans = (ans + (A[i] - A[n - i - 1]) * p) % kMod;

p = (p << 1) % kMod;

}

return (ans + kMod) % kMod;

}

private:

void countingSort(vector<int>& A) {

int max_v = INT_MIN;

int min_v = INT_MAX;

for (int a : A) {

if (a > max_v) max_v = a;

if (a < min_v) min_v = a;

}

vector<int> c(max_v - min_v + 1);

for (int a : A)

++c[a - min_v];

int i = 0;

int j = 0;

while (i < c.size()) {

if (--c[i] >= 0)

A[j++] = i + min_v;

else

++i;

}

};

花花酱 LeetCode 628. Maximum Product of Three Numbers

By zxi on August 17, 2018

Problem

Given an integer array, find three numbers whose product is maximum and output the maximum product.

Example 1:

Input: [1,2,3]
Output: 6

Example 2:

Input: [1,2,3,4]
Output: 24

Note:

The length of the given array will be in range [3,10⁴] and all elements are in the range [-1000, 1000].
Multiplication of any three numbers in the input won’t exceed the range of 32-bit signed integer.

Idea:

Find the top 3 numbers t1, t2, t3, and bottom 2 numbers, b1, b2.

If all numbers are positives, answer must be t1 * t2 * t3.

Since the number can go negative, the answer must be either t1*t2*t3 or b1 * b2 * t1, if b1 and b2 are both negatives.

ex. nums: [5, 1, -6, 3, -1]

t1, t2, t3: 5, 3, 1

b1, b2: -6, -1

t1 * t2 * t3 = 15

t1 * b1 * b2 = 30

Solution 1: Manual Tracking

Time complexity: O(n)

Space complexity: O(1)

// Author: Huahua

// Running time: 28 ms

class Solution {

public:

int maximumProduct(vector<int>& nums) {

int max1 = INT_MIN;

int max2 = INT_MIN;

int max3 = INT_MIN;

int min1 = INT_MAX;

int min2 = INT_MAX;

for (const int num: nums) {

if (num > max1) {

max3 = max2;

max2 = max1;

max1 = num;

} else if (num > max2) {

max3 = max2;

max2 = num;

} else if (num > max3) {

max3=num;

}

if (num < min1) {

min2 = min1;

min1 = num;

} else if (num < min2) {

min2=num;

}

return max(max1*max2*max3, max1*min1*min2);

}

};

Solution 2: Sorting

Time complexity: O(nlogn)

Space complexity: O(1)

// Author: Huahua

// Running time: 48 ms

class Solution {

public:

int maximumProduct(vector<int>& nums) {

int n = nums.size();

sort(nums.rbegin(), nums.rend());

return max(nums[0] * nums[1] * nums[2], nums[0] * nums[n - 1] * nums[n - 2]);

}

};

Solution 3: Two Heaps (Priority Queues)

Time complexity: O(nlog3)

Space complexity: O(2 + 3)

// Author: Huahua

// Running time: 40 ms

class Solution {

public:

int maximumProduct(vector<int>& nums) {

priority_queue<int, vector<int>, less<int>> min_q; // max_heap

priority_queue<int, vector<int>, greater<int>> max_q; // min_heap

for (int num : nums) {

max_q.push(num);

if (max_q.size() > 3) max_q.pop();

min_q.push(num);

if (min_q.size() > 2) min_q.pop();

}

int max3 = max_q.top(); max_q.pop();

int max2 = max_q.top(); max_q.pop();

int max1 = max_q.top(); max_q.pop();

int min2 = min_q.top(); min_q.pop();

int min1 = min_q.top(); min_q.pop();

return max(max1 * max2 * max3, max1 * min1 * min2);

}

};

花花酱 LeetCode 463. Island Perimeter

By zxi on August 17, 2018

Problem

You are given a map in form of a two-dimensional integer grid where 1 represents land and 0 represents water. Grid cells are connected horizontally/vertically (not diagonally). The grid is completely surrounded by water, and there is exactly one island (i.e., one or more connected land cells). The island doesn’t have “lakes” (water inside that isn’t connected to the water around the island). One cell is a square with side length 1. The grid is rectangular, width and height don’t exceed 100. Determine the perimeter of the island.

Example:

[[0,1,0,0],
 [1,1,1,0],
 [0,1,0,0],
 [1,1,0,0]]

Answer: 16
Explanation: The perimeter is the 16 yellow stripes in the image below:

Solution: Counting

If a land has 0 neighbour, it contributes 4 to the perimeter

If a land has 1 neighbour, it contributes 3 to the perimeter

If a land has 2 neighbours, it contributes 2 to the perimeter

If a land has 3 neighbours, it contributes 1 to the perimeter

If a land has 4 neighbours, it contributes 0 to the perimeter

perimeter = area * 4 – total_neighbours

Time complexity: O(mn)

Space complexity: O(1)

// Author: Huahua

// Running time: 148 ms

class Solution {

public:

int islandPerimeter(vector<vector<int>>& grid) {

if (grid.empty()) return 0;

int m = grid.size();

int n = grid[0].size();

int area = 0;

int conn = 0;

for (int y = 0; y < m; ++y)

for (int x = 0; x < n; ++x)

if (grid[y][x] == 1) {

++area;

if (y > 0 && grid[y - 1][x] == 1) ++conn;

if (x > 0 && grid[y][x - 1] == 1) ++conn;

}

return area*4 - conn*2;

}

};