Posts published in August 2018

花花酱 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 890. Find and Replace Pattern

By zxi on August 19, 2018

Problem

You have a list of words and a pattern, and you want to know which words in words matches the pattern.

A word matches the pattern if there exists a permutation of letters p so that after replacing every letter x in the pattern with p(x), we get the desired word.

(Recall that a permutation of letters is a bijection from letters to letters: every letter maps to another letter, and no two letters map to the same letter.)

Return a list of the words in words that match the given pattern.

You may return the answer in any order.

Example 1:

Input: words = ["abc","deq","mee","aqq","dkd","ccc"], pattern = "abb"
Output: ["mee","aqq"]
Explanation: "mee" matches the pattern because there is a permutation {a -> m, b -> e, ...}. 
"ccc" does not match the pattern because {a -> c, b -> c, ...} is not a permutation,
since a and b map to the same letter.

Note:

1 <= words.length <= 50
1 <= pattern.length = words[i].length <= 20

Solution: Remember the last pos of each char.

Time complexity: O(n*l)

Space complexity: O(128) -> O(26)

C++

// Author: Huahua

// Running time: 0 ms

class Solution {

public:

vector<string> findAndReplacePattern(vector<string>& words, string pattern) {

vector<string> ans;

for (const string& w : words)

if (match(w, pattern)) ans.push_back(w);

return ans;

}

private:

bool match(const string& w, const string& p) {

vector<int> last_pos_w(128, INT_MIN); // last pos of x in w

vector<int> last_pos_p(128, INT_MIN); // last pos of x in p

for (int i = 0; i < w.size(); ++i)

if (last_pos_w[w[i]] != last_pos_p[p[i]]) {

return false;

} else {

last_pos_w[w[i]] = last_pos_p[p[i]] = i;

}

return true;

}

};

花花酱 LeetCode 888. Fair Candy Swap

By zxi on August 19, 2018

Problem

Alice and Bob have candy bars of different sizes: A[i] is the size of the i-th bar of candy that Alice has, and B[j] is the size of the j-th bar of candy that Bob has.

Since they are friends, they would like to exchange one candy bar each so that after the exchange, they both have the same total amount of candy. (The total amount of candy a person has is the sum of the sizes of candy bars they have.)

Return an integer array ans where ans[0] is the size of the candy bar that Alice must exchange, and ans[1] is the size of the candy bar that Bob must exchange.

If there are multiple answers, you may return any one of them. It is guaranteed an answer exists.

Example 1:

Input: A = [1,1], B = [2,2]
Output: [1,2]

Example 2:

Input: A = [1,2], B = [2,3]
Output: [1,2]

Example 3:

Input: A = [2], B = [1,3]
Output: [2,3]

Example 4:

Input: A = [1,2,5], B = [2,4]
Output: [5,4]

Note:

1 <= A.length <= 10000
1 <= B.length <= 10000
1 <= A[i] <= 100000
1 <= B[i] <= 100000
It is guaranteed that Alice and Bob have different total amounts of candy.
It is guaranteed there exists an answer.

Solution: HashTable

Time complexity: O(A+B)

Space complexity: O(B)

Clean version

// Author: Huahua

// Running time: 132 ms

class Solution {

public:

vector<int> fairCandySwap(vector<int>& A, vector<int>& B) {

int sum_a = accumulate(begin(A), end(A), 0);

int sum_b = accumulate(begin(B), end(B), 0);

unordered_set<int> s(begin(B), end(B));

int diff = (sum_b - sum_a) / 2;

for (int a : A)

if (s.count(a + diff)) return {a, a + diff};

}

};

Faster version

// Author: Huahua

// Running time: 68 ms

class Solution {

public:

vector<int> fairCandySwap(vector<int>& A, vector<int>& B) {

int sum_a = accumulate(begin(A), end(A), 0);

int sum_b = 0;

unordered_set<int> s;

for (int b : B) {

sum_b += b;

s.insert(b);

}

int diff = (sum_b - sum_a) / 2;

for (int a : A)

if (s.count(a + diff)) return {a, a + diff};

}

};

花花酱 LeetCode 617. Merge Two Binary Trees

By zxi on August 17, 2018

Problem

Given two binary trees and imagine that when you put one of them to cover the other, some nodes of the two trees are overlapped while the others are not.

You need to merge them into a new binary tree. The merge rule is that if two nodes overlap, then sum node values up as the new value of the merged node. Otherwise, the NOT null node will be used as the node of new tree.

Example 1:

Input: 
	Tree 1                     Tree 2                  
          1                         2                             
         / \                       / \                            
        3   2                     1   3                        
       /                           \   \                      
      5                             4   7                  
Output: 
Merged tree:
	     3
	    / \
	   4   5
	  / \   \ 
	 5   4   7

Note: The merging process must start from the root nodes of both trees.

Solution: Recursion

Reuse t1/t2

// Author: Huahua

// Running time: 52 ms

class Solution {

public:

TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {

if (t1 == nullptr) return t2;

if (t2 == nullptr) return t1;

auto root = t1;

root->val += t2->val;

root->left = mergeTrees(t1->left, t2->left);

root->right = mergeTrees(t1->right, t2->right);

return root;

}

};

Create a copy

// Author: Huahua

// Running time: 52 ms

class Solution {

public:

TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {

if (t1 == nullptr) return copyTree(t2);

if (t2 == nullptr) return copyTree(t1);

TreeNode* root = new TreeNode(t1->val + t2->val);

root->left = mergeTrees(t1->left, t2->left);

root->right = mergeTrees(t1->right, t2->right);

return root;

}

private:

TreeNode* copyTree(TreeNode* root) {

if (root == nullptr) return nullptr;

TreeNode* r = new TreeNode(root->val);

r->left = copyTree(root->left);

r->right = copyTree(root->right);

return r;

}

};