Posts tagged as “hard”

花花酱 LeetCode 749. Contain Virus

By zxi on December 18, 2017

题目大意：给你一个格子地图上面有一些病毒用1表示，未受感染的格子用0表示。带有病毒的格子每天会向四周的格子传播病毒。在每一天，你必须在最大的病毒（连通区域）的周围建造墙阻挡病毒传播。问你一共需要多少个墙才能阻挡所有病毒传播。

Problems:

A virus is spreading rapidly, and your task is to quarantine the infected area by installing walls.

The world is modeled as a 2-D array of cells, where 0 represents uninfected cells, and 1 represents cells contaminated with the virus. A wall (and only one wall) can be installed between any two 4-directionally adjacent cells, on the shared boundary.

Every night, the virus spreads to all neighboring cells in all four directions unless blocked by a wall. Resources are limited. Each day, you can install walls around only one region — the affected area (continuous block of infected cells) that threatens the most uninfected cells the following night. There will never be a tie.

Can you save the day? If so, what is the number of walls required? If not, and the world becomes fully infected, return the number of walls used.

Example 1:

Input: grid = 
[[0,1,0,0,0,0,0,1],
 [0,1,0,0,0,0,0,1],
 [0,0,0,0,0,0,0,1],
 [0,0,0,0,0,0,0,0]]
Output: 10
Explanation:
There are 2 contaminated regions.
On the first day, add 5 walls to quarantine the viral region on the left. The board after the virus spreads is:

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

On the second day, add 5 walls to quarantine the viral region on the right. The virus is fully contained.

Example 2:

Input: grid = 
[[1,1,1],
 [1,0,1],
 [1,1,1]]
Output: 4
Explanation: Even though there is only one cell saved, there are 4 walls built.
Notice that walls are only built on the shared boundary of two different cells.

Example 3:

Input: grid = 
[[1,1,1,0,0,0,0,0,0],
 [1,0,1,0,1,1,1,1,1],
 [1,1,1,0,0,0,0,0,0]]
Output: 13
Explanation: The region on the left only builds two new walls.

Note:

The number of rows and columns of grid will each be in the range [1, 50].
Each grid[i][j] will be either 0 or 1.
Throughout the described process, there is always a contiguous viral region that will infect strictly more uncontaminated squares in the next round.

Idea:

Use DFS to find virus regions, next affected regions and # of walls needed to block each virus region.

Simulate the virus expansion process.

Solution:

C++ / DFS

Time complexity: O(n^3)

Space complexity: O(n^2)

// Author: Huahua

// Runtime: 12 ms

class Solution {

public:

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

int m = grid.size();

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

int total_walls = 0;

while (true) {

vector<int> visited(m * n, 0);

vector<int> virus_area;

vector<unordered_set<int>> nexts;

int block_index = -1;

int block_walls = -1;

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

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

int key = i * n + j;

if (grid[i][j] != 1 || visited[key]) continue;

vector<int> curr;

unordered_set<int> next;

int walls = 0;

getArea(j, i, m, n, grid, visited, curr, next, walls);

if (next.empty()) continue;

if (nexts.empty() || next.size() > nexts[block_index].size()) {

virus_area.swap(curr);

block_index = nexts.size();

block_walls = walls;

}

nexts.push_back(std::move(next));

}

if (nexts.empty()) break;

total_walls += block_walls;

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

if (i == block_index) {

for (const int key : virus_area) {

int y = key / n;

int x = key % n;

grid[y][x] = 2; // blocked.

}

} else {

for (const int key : nexts[i]) {

int y = key / n;

int x = key % n;

grid[y][x] = 1; // newly affected

}

return total_walls;

}

private:

void getArea(

int x, int y, int m, int n,

vector<vector<int>>& grid,

vector<int>& visited,

vector<int>& curr,

unordered_set<int>& next,

int& walls) {

static int dirs[4][2] = {{0, 1}, {0, -1}, {1, 0}, {-1, 0}};

if (x < 0 || x >= n || y < 0 || y >= m || grid[y][x] == 2) return;

int key = y * n + x;

// need one wall

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

++walls;

next.insert(key);

return;

}

if (visited[key]) return;

visited[key] = 1;

curr.push_back(key);

for (int i = 0; i < 4; ++i)

getArea(x + dirs[i][0], y + dirs[i][1], m, n, grid, visited, curr, next, walls);

}

};

Solution 1: Count Boundaries

Time complexity: O(n^2)

Space complexity: O(n)

C++

// Author: Huahua

// Runtime: 116 ms

class MyCalendarThree {

public:

MyCalendarThree() {}

int book(int start, int end) {

++deltas_[start];

--deltas_[end];

int ans = 0;

int curr = 0;

for (const auto& kv : deltas_)

ans = max(ans, curr += kv.second);

return ans;

}

private:

map<int, int> deltas_;

};

Solution 2

C++

// Author: Huahua

// Runtime: 66 ms

class MyCalendarThree {

public:

MyCalendarThree() {

counts_[INT_MIN] = 0;

counts_[INT_MAX] = 0;

max_count_ = 0;

}

int book(int start, int end) {

auto l = prev(counts_.upper_bound(start)); // l->first < start

auto r = counts_.lower_bound(end); // r->first >= end

for (auto curr = l, next = curr; curr != r; curr = next) {

++next;

if (next->first > end)

counts_[end] = curr->second;

if (curr->first <= start && next->first > start)

max_count_ = max(max_count_, counts_[start] = curr->second + 1);

else

max_count_ = max(max_count_, ++curr->second);

}

return max_count_;

}

private:

map<int, int> counts_;

int max_count_;

};

Solution 3: Segment Tree

C++

// Author: Huahua

// Runtime: 63 ms (<95.65%)

class MyCalendarThree {

public:

MyCalendarThree(): max_count_(0) {

root_.reset(new Node(0, 100000000, 0));

}

int book(int start, int end) {

Add(start, end, root_.get());

return max_count_;

}

private:

struct Node {

Node(int l, int r, int count):l(l), m(-1), r(r), count(count){}

int l;

int m;

int r;

int count;

std::unique_ptr<Node> left;

std::unique_ptr<Node> right;

};

void Add(int start, int end, Node* root) {

if (root->m != -1) {

if (end <= root->m)

Add(start, end, root->left.get());

else if(start >= root->m)

Add(start, end, root->right.get());

else {

Add(start, root->m, root->left.get());

Add(root->m, end, root->right.get());

}

return;

}

if (start == root->l && end == root->r)

max_count_ = max(max_count_, ++root->count);

else if (start == root->l) {

root->m = end;

root->left.reset(new Node(start, root->m, root->count + 1));

root->right.reset(new Node(root->m, root->r, root->count));

max_count_ = max(max_count_, root->count + 1);

} else if(end == root->r) {

root->m = start;

root->left.reset(new Node(root->l, root->m, root->count));

root->right.reset(new Node(root->m, root->r, root->count + 1));

max_count_ = max(max_count_, root->count + 1);

} else {

root->m = start;

root->left.reset(new Node(root->l, root->m, root->count));

root->right.reset(new Node(root->m, root->r, root->count));

Add(start, end, root->right.get());

}

std::unique_ptr<Node> root_;

int max_count_;

};

Python3

"""

Author: Huahua

Runtime: 436 ms (<88.88%)

"""

class Node:

def __init__(self, l, r, count):

self.l = l

self.m = -1

self.r = r

self.count = count

self.left = None

self.right = None

class MyCalendarThree:

def __init__(self):

self.root = Node(0, 10**9, 0)

self.max = 0

def book(self, start, end):

self.add(start, end, self.root)

return self.max

def add(self, start, end, root):

if root.m != -1:

if end <= root.m: self.add(start, end, root.left)

elif start >= root.m: self.add(start, end, root.right)

else:

self.add(start, root.m, root.left)

self.add(root.m, end, root.right)

return

if start == root.l and end == root.r:

root.count += 1

self.max = max(self.max, root.count)

elif start == root.l:

root.m = end

root.left = Node(start, root.m, root.count + 1)

root.right = Node(root.m, root.r, root.count)

self.max = max(self.max, root.count + 1)

elif end == root.r:

root.m = start;

root.left = Node(root.l, root.m, root.count)

root.right = Node(root.m, root.r, root.count + 1)

self.max = max(self.max, root.count + 1)

else:

root.m = start

root.left = Node(root.l, root.m, root.count)

root.right = Node(root.m, root.r, root.count)

self.add(start, end, root.right)

Related Problems:

花花酱 LeetCode 741. Cherry Pickup

By zxi on December 4, 2017

题目大意：给你樱桃田的地图（1: 樱桃, 0: 空, -1: 障碍物）。然你从左上角走到右下角（只能往右或往下），再从右下角走回左上角（只能往左或者往上）。问你最多能采到多少棵樱桃。

Problem:

In a N x N grid representing a field of cherries, each cell is one of three possible integers.

0 means the cell is empty, so you can pass through;
1 means the cell contains a cherry, that you can pick up and pass through;
-1 means the cell contains a thorn that blocks your way.

Your task is to collect maximum number of cherries possible by following the rules below:

Starting at the position (0, 0) and reaching (N-1, N-1) by moving right or down through valid path cells (cells with value 0 or 1);
After reaching (N-1, N-1), returning to (0, 0) by moving left or up through valid path cells;
When passing through a path cell containing a cherry, you pick it up and the cell becomes an empty cell (0);
If there is no valid path between (0, 0) and (N-1, N-1), then no cherries can be collected.

Example 1:

Input: grid =

[[0, 1, -1],

[1, 0, -1],

[1, 1, 1]]

Output: 5

Explanation:

The player started at (0, 0) and went down, down, right right to reach (2, 2).

4 cherries were picked up during this single trip, and the matrix becomes [[0,1,-1],[0,0,-1],[0,0,0]].

Then, the player went left, up, up, left to return home, picking up one more cherry.

The total number of cherries picked up is 5, and this is the maximum possible.

Note:

grid is an N by N 2D array, with 1 <= N <= 50.
Each grid[i][j] is an integer in the set {-1, 0, 1}.
It is guaranteed that grid[0][0] and grid[N-1][N-1] are not -1.

Idea:

Key observation: (0,0) to (n-1, n-1) to (0, 0) is the same as (n-1,n-1) to (0,0) twice

Two people starting from (n-1, n-1) and go to (0, 0).
They move one step (left or up) at a time simultaneously. And pick up the cherry within the grid (if there is one).
if they ended up at the same grid with a cherry. Only one of them can pick up it.

Solution: DP / Recursion with memoization.

x1, y1, x2 to represent a state y2 can be computed: y2 = x1 + y1 – x2

dp(x1, y1, x2) computes the max cherries if start from {(x1, y1), (x2, y2)} to (0, 0), which is a recursive function.

Since two people move independently, there are 4 subproblems: (left, left), (left, up), (up, left), (left, up). Finally, we have:

dp(x1, y1, x2)= g[y1][x1] + g[y2][x2] + max{dp(x1-1,y1,x2-1), dp(x1,y1-1,x2-1), dp(x1-1,y1,x2), dp(x1,y1-1,x2)}

Time complexity: O(n^3)

Space complexity: O(n^3)

Solution: DP

Time complexity: O(n^3)

Space complexity: O(n^3)

C ++

// Author: Huahua

// Runtime: 32 ms

class Solution {

public:

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

const int n = grid.size();

grid_ = &grid;

m_ = vector<vector<vector<int>>>(

n, vector<vector<int>>(n, vector<int>(n, INT_MIN)));

return max(0, dp(n - 1, n - 1, n - 1));

}

private:

// max cherries from (x1, y1) to (0, 0) + (x2, y2) to (0, 0)

int dp(int x1, int y1, int x2) {

const int y2 = x1 + y1 - x2;

if (x1 < 0 || y1 < 0 || x2 < 0 || y2 < 0) return -1;

if ((*grid_)[y1][x1] < 0 || (*grid_)[y2][x2] < 0) return -1;

if (x1 == 0 && y1 == 0) return (*grid_)[y1][x1];

if (m_[x1][y1][x2] != INT_MIN) return m_[x1][y1][x2];

int ans = max(max(dp(x1 - 1, y1, x2 - 1), dp(x1, y1 - 1, x2)),

max(dp(x1, y1 - 1, x2 - 1), dp(x1 - 1, y1, x2)));

if (ans < 0) return m_[x1][y1][x2] = -1;

ans += (*grid_)[y1][x1];

if (x1 != x2) ans += (*grid_)[y2][x2];

return m_[x1][y1][x2] = ans;

}

vector<vector<vector<int>>> m_;

vector<vector<int>>* grid_; // does not own the object

};

Java

// Author: Huahua

class Solution {

private int[][] grid;

private int[][][] memo;

public int cherryPickup(int[][] grid) {

this.grid = grid;

int m = grid.length;

int n = grid[0].length;

memo = new int[m][n][n];

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

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

Arrays.fill(memo[i][j], Integer.MIN_VALUE);

return Math.max(0, dp(n - 1, m - 1, n - 1));

}

private int dp(int x1, int y1, int x2) {

final int y2 = x1 + y1 - x2;

if (x1 < 0 || y1 < 0 || x2 < 0 || y2 < 0) return -1;

if (grid[y1][x1] < 0 || grid[y2][x2] < 0) return -1;

if (x1 == 0 && y1 == 0) return grid[y1][x1];

if (memo[y1][x1][x2] != Integer.MIN_VALUE) return memo[y1][x1][x2];

memo[y1][x1][x2] = Math.max(Math.max(dp(x1 - 1, y1, x2 - 1), dp(x1, y1 - 1, x2)),

Math.max(dp(x1, y1 - 1, x2 - 1), dp(x1 - 1, y1, x2)));

if (memo[y1][x1][x2] >= 0) {

memo[y1][x1][x2] += grid[y1][x1];

if (x1 != x2) memo[y1][x1][x2] += grid[y2][x2];

}

return memo[y1][x1][x2];

}

Related Problems:

花花酱 LeetCode 399. Evaluate Division

By zxi on December 1, 2017

题目大意：给你一些含有变量名的分式的值，让你计算另外一些分式的值，如果不能计算返回-1。

Problem:

Equations are given in the format A / B = k, where A and B are variables represented as strings, and k is a real number (floating point number). Given some queries, return the answers. If the answer does not exist, return -1.0.

Example:
Given a / b = 2.0, b / c = 3.0.
queries are: a / c = ?, b / a = ?, a / e = ?, a / a = ?, x / x = ? .
return [6.0, 0.5, -1.0, 1.0, -1.0 ].

The input is: vector<pair<string, string>> equations, vector<double>& values, vector<pair<string, string>> queries , where equations.size() == values.size(), and the values are positive. This represents the equations. Return vector<double>.

According to the example above:

equations = [ ["a", "b"], ["b", "c"] ],

values = [2.0, 3.0],

queries = [ ["a", "c"], ["b", "a"], ["a", "e"], ["a", "a"], ["x", "x"] ].

The input is always valid. You may assume that evaluating the queries will result in no division by zero and there is no contradiction.

Solution 1: DFS

C++

// Author: Huahua

// Runtime: 3 ms

class Solution {

public:

vector<double> calcEquation(vector<pair<string, string>> equations, vector<double>& values, vector<pair<string, string>> queries) {

// g[A][B] = k -> A / B = k

unordered_map<string, unordered_map<string, double>> g;

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

const string& A = equations[i].first;

const string& B = equations[i].second;

const double k = values[i];

g[A][B] = k;

g[B][A] = 1.0 / k;

}

vector<double> ans;

for (const auto& pair : queries) {

const string& X = pair.first;

const string& Y = pair.second;

if (!g.count(X) || !g.count(Y)) {

ans.push_back(-1.0);

continue;

}

unordered_set<string> visited;

ans.push_back(divide(X, Y, g, visited));

}

return ans;

}

private:

// get result of A / B

double divide(const string& A, const string& B,

unordered_map<string, unordered_map<string, double>>& g,

unordered_set<string>& visited) {

if (A == B) return 1.0;

visited.insert(A);

for (const auto& pair : g[A]) {

const string& C = pair.first;

if (visited.count(C)) continue;

double d = divide(C, B, g, visited); // d = C / B

// A / B = C / B * A / C

if (d > 0) return d * g[A][C];

}

return -1.0;

}

};

Java

// Updated: 2020/9/27

// Author: Huahua

// Running time: 1 ms

class Solution {

private Map<String, HashMap<String, Double>> g = new HashMap<>();

public double[] calcEquation(List<List<String>> equations, double[] values, List<List<String>> queries) {

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

String x = equations.get(i).get(0);

String y = equations.get(i).get(1);

double k = values[i];

g.computeIfAbsent(x, l -> new HashMap<String, Double>()).put(y, k);

g.computeIfAbsent(y, l -> new HashMap<String, Double>()).put(x, 1.0 / k);

}

double[] ans = new double[queries.size()];

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

String x = queries.get(i).get(0);

String y = queries.get(i).get(1);

if (!g.containsKey(x) || !g.containsKey(y)) {

ans[i] = -1.0;

} else {

ans[i] = divide(x, y, new HashSet<String>());

}

return ans;

}

private double divide(String x, String y, Set<String> visited) {

if (x.equals(y)) return 1.0;

visited.add(x);

if (!g.containsKey(x)) return -1.0;

for (String n : g.get(x).keySet()) {

if (visited.contains(n)) continue;

visited.add(n);

double d = divide(n, y, visited);

if (d > 0) return d * g.get(x).get(n);

}

return -1.0;

}

Python3

"""

Author: Huahua

Running time: 32 ms (beats 100%)

"""

class Solution:

def calcEquation(self, equations, values, queries):

def divide(x, y, visited):

if x == y: return 1.0

visited.add(x)

for n in g[x]:

if n in visited: continue

visited.add(n)

d = divide(n, y, visited)

if d > 0: return d * g[x][n]

return -1.0

g = collections.defaultdict(dict)

for (x, y), v in zip(equations, values):

g[x][y] = v

g[y][x] = 1.0 / v

ans = [divide(x, y, set()) if x in g and y in g else -1 for x, y in queries]

return ans

Solution 2: Union Find

C++

// Author: Huahua

// Runtime: 3 ms

class Solution {

public:

vector<double> calcEquation(const vector<pair<string, string>>& equations, vector<double>& values, const vector<pair<string, string>>& queries) {

// parents["A"] = {"B", 2.0} -> A = 2.0 * B

// parents["B"] = {"C", 3.0} -> B = 3.0 * C

unordered_map<string, pair<string, double>> parents;

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

const string& A = equations[i].first;

const string& B = equations[i].second;

const double k = values[i];

// Neighter is in the forrest

if (!parents.count(A) && !parents.count(B)) {

parents[A] = {B, k};

parents[B] = {B, 1.0};

} else if (!parents.count(A)) {

parents[A] = {B, k};

} else if (!parents.count(B)) {

parents[B] = {A, 1.0 / k};

} else {

auto& rA = find(A, parents);

auto& rB = find(B, parents);

parents[rA.first] = {rB.first, k / rA.second * rB.second};

}

vector<double> ans;

for (const auto& pair : queries) {

const string& X = pair.first;

const string& Y = pair.second;

if (!parents.count(X) || !parents.count(Y)) {

ans.push_back(-1.0);

continue;

}

auto& rX = find(X, parents); // {rX, X / rX}

auto& rY = find(Y, parents); // {rY, Y / rY}

if (rX.first != rY.first)

ans.push_back(-1.0);

else // X / Y = (X / rX / (Y / rY))

ans.push_back(rX.second / rY.second);

}

return ans;

}

private:

pair<string, double>& find(const string& C, unordered_map<string, pair<string, double>>& parents) {

if (C != parents[C].first) {

const auto& p = find(parents[C].first, parents);

parents[C].first = p.first;

parents[C].second *= p.second;

}

return parents[C];

}

};

Java

// Updated: 9/27/2020

// Author: Huahua

// Running time: 3 ms

class Solution {

class Node {

public String parent;

public double ratio;

public Node(String parent, double ratio) {

this.parent = parent;

this.ratio = ratio;

}

class UnionFindSet {

private Map<String, Node> parents = new HashMap<>();

public Node find(String s) {

if (!parents.containsKey(s)) return null;

Node n = parents.get(s);

if (!n.parent.equals(s)) {

Node p = find(n.parent);

n.parent = p.parent;

n.ratio *= p.ratio;

}

return n;

}

public void union(String A, String B, double ratio) {

boolean hasA = parents.containsKey(A);

boolean hasB = parents.containsKey(B);

if (!hasA && !hasB) {

parents.put(A, new Node(B, ratio));

parents.put(B, new Node(B, 1.0));

} else if (!hasA) {

parents.put(A, new Node(B, ratio));

} else if (!hasB) {

parents.put(B, new Node(A, 1.0 / ratio));

} else {

Node rA = find(A);

Node rB = find(B);

parents.put(rA.parent,

new Node(rB.parent, ratio / rA.ratio * rB.ratio));

}

public double[] calcEquation(List<List<String>> equations, double[] values, List<List<String>> queries) {

UnionFindSet u = new UnionFindSet();

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

u.union(equations.get(i).get(0), equations.get(i).get(1), values[i]);

double[] ans = new double[queries.size()];

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

Node rx = u.find(queries.get(i).get(0));

Node ry = u.find(queries.get(i).get(1));

if (rx == null || ry == null || !rx.parent.equals(ry.parent))

ans[i] = -1.0;

else

ans[i] = rx.ratio / ry.ratio;

}

return ans;

}

Python3

"""

Author: Huahua

Running time: 32 ms (beats 100%)

"""

class Solution:

def calcEquation(self, equations, values, queries):

def find(x):

if x != U[x][0]:

px, pv = find(U[x][0])

U[x] = (px, U[x][1] * pv)

return U[x]

def divide(x, y):

rx, vx = find(x)

ry, vy = find(y)

if rx != ry: return -1.0

return vx / vy

U = {}

for (x, y), v in zip(equations, values):

if x not in U and y not in U:

U[x] = (y, v)

U[y] = (y, 1.0)

elif x not in U:

U[x] = (y, v)

elif y not in U:

U[y] = (x, 1.0 / v)

else:

rx, vx = find(x)

ry, vy = find(y)

U[rx] = (ry, v / vx * vy)

ans = [divide(x, y) if x in U and y in U else -1 for x, y in queries]

return ans

Related Problems:

花花酱 LeetCode 730. Count Different Palindromic Subsequences

By zxi on November 20, 2017

Problem:

Given a string S, find the number of different non-empty palindromic subsequences in S, and return that number modulo 10^9 + 7.

A subsequence of a string S is obtained by deleting 0 or more characters from S.

A sequence is palindromic if it is equal to the sequence reversed.

Two sequences A_1, A_2, ... and B_1, B_2, ... are different if there is some i for which A_i != B_i.

Example 1:

Input: 
S = 'bccb'
Output: 6
Explanation: 
The 6 different non-empty palindromic subsequences are 'b', 'c', 'bb', 'cc', 'bcb', 'bccb'.
Note that 'bcb' is counted only once, even though it occurs twice.

Example 2:

Input: 
S = 'abcdabcdabcdabcdabcdabcdabcdabcddcbadcbadcbadcbadcbadcbadcbadcba'
Output: 104860361
Explanation: 
There are 3104860382 different non-empty palindromic subsequences, which is 104860361 modulo 10^9 + 7.

Note:

The length of S will be in the range [1, 1000].
Each character S[i] will be in the set {'a', 'b', 'c', 'd'}.

Idea:

Solution 1: Recursion with memoization

Time complexity: O(n^2)
Space complexity: O(n^2)

C++

// Author: Huahua

// Runtime: 72 ms

class Solution {

public:

int countPalindromicSubsequences(const string& S) {

const int n = S.length();

m_ = vector<vector<int>>(n + 1, vector<int>(n + 1, 0));

return count(S, 0, S.length() - 1);

}

private:

static constexpr long kMod = 1000000007;

long count(const string& S, int s, int e) {

if (s > e) return 0;

if (s == e) return 1;

if (m_[s][e] > 0) return m_[s][e];

long ans = 0;

if (S[s] == S[e]) {

int l = s + 1;

int r = e - 1;

while (l <= r && S[l] != S[s]) ++l;

while (l <= r && S[r] != S[s]) --r;

if (l > r)

ans = count(S, s + 1, e - 1) * 2 + 2;

else if (l == r)

ans = count(S, s + 1, e - 1) * 2 + 1;

else

ans = count(S, s + 1, e - 1) * 2

- count(S, l + 1, r - 1);

} else {

ans = count(S, s, e - 1)

+ count(S, s + 1, e)

- count(S, s + 1, e - 1);

}

return m_[s][e] = (ans + kMod) % kMod;

}

vector<vector<int>> m_;

};

Python

"""

Author: Huahua

Runtime: 3582 ms

"""

class Solution:

def countPalindromicSubsequences(self, S):

def count(S, i, j):

if i > j: return 0

if i == j: return 1

if self.m_[i][j]: return self.m_[i][j]

if S[i] == S[j]:

ans = count(S, i + 1, j - 1) * 2

l = i + 1

r = j - 1

while l <= r and S[l] != S[i]: l += 1

while l <= r and S[r] != S[i]: r -= 1

if l > r: ans += 2

elif l == r: ans += 1

else: ans -= count(S, l + 1, r - 1)

else:

ans = count(S, i + 1, j) + count(S, i, j - 1) - count(S, i + 1, j - 1)

self.m_[i][j] = ans % 1000000007

return self.m_[i][j]

n = len(S)

self.m_ = [[None for _ in range(n)] for _ in range(n)]

return count(S, 0, n - 1)

Java

// Author: Huahua

// Runtime: 107 ms

class Solution {

private int[][] m_;

private static final int kMod = 1000000007;

public int countPalindromicSubsequences(String S) {

int n = S.length();

m_ = new int[n][n];

return count(S.toCharArray(), 0, n - 1);

}

private int count(char[] s, int i, int j) {

if (i > j) return 0;

if (i == j) return 1;

if (m_[i][j] > 0) return m_[i][j];

long ans = 0;

if (s[i] == s[j]) {

ans += count(s, i + 1, j - 1) * 2;

int l = i + 1;

int r = j - 1;

while (l <= r && s[l] != s[i]) ++l;

while (l <= r && s[r] != s[i]) --r;

if (l > r) ans += 2;

else if (l == r) ans += 1;

else ans -= count(s, l + 1, r - 1);

} else {

ans = count(s, i, j - 1)

+ count(s, i + 1, j)

- count(s, i + 1, j - 1);

}

m_[i][j] = (int)((ans + kMod) % kMod);

return m_[i][j];

}

Solution 2: DP

C++

// Author: Huahua

// Runtime: 79 ms

class Solution {

public:

int countPalindromicSubsequences(const string& S) {

int n = S.length();

vector<vector<int>> dp(n, vector<int>(n, 0));

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

dp[i][i] = 1;

for (int len = 1; len <= n; ++len) {

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

const int j = i + len;

if (S[i] == S[j]) {

dp[i][j] = dp[i + 1][j - 1] * 2;

int l = i + 1;

int r = j - 1;

while (l <= r && S[l] != S[i]) ++l;

while (l <= r && S[r] != S[i]) --r;

if (l == r) dp[i][j] += 1;

else if (l > r) dp[i][j] += 2;

else dp[i][j] -= dp[l + 1][r - 1];

} else {

dp[i][j] = dp[i][j - 1] + dp[i + 1][j] - dp[i + 1][j - 1];

}

dp[i][j] = (dp[i][j] + kMod) % kMod;

}

return dp[0][n - 1];

}

private:

static constexpr long kMod = 1000000007;

};

Pyhton

"""

Author: Huahua

Runtime: 3639 ms

"""

class Solution:

def countPalindromicSubsequences(self, S):

n = len(S)

if n == 0: return 0

dp = [[0 for _ in range(n)] for _ in range(n)]

for i in range(n): dp[i][i] = 1

for size in range(2, n + 1):

for i in range(n - size + 1):

j = i + size - 1

if S[i] == S[j]:

dp[i][j] = dp[i + 1][j - 1] * 2

l = i + 1

r = j - 1

while l <= r and S[l] != S[i]: l += 1

while l <= r and S[r] != S[i]: r -= 1

if l > r: dp[i][j] += 2

elif l == r: dp[i][j] += 1

else: dp[i][j] -= dp[l + 1][r - 1]

else:

dp[i][j] = dp[i + 1][j] + dp[i][j - 1] - dp[i + 1][j - 1]

dp[i][j] %= 1000000007

return dp[0][n - 1]

Related Problems:

[解题报告] LeetCode 664. Strange Printer

Posts tagged as “hard”

花花酱 LeetCode 749. Contain Virus

Problems:

Example 1:

Example 2:

Example 3:

Note:

Idea:

Solution:

Related Problems:

花花酱 LeetCode 732. My Calendar III

Solution 1: Count Boundaries

C++

Solution 2

C++

Solution 3: Segment Tree

C++

Python3

花花酱 LeetCode 741. Cherry Pickup

Solution: DP

C ++

Java

花花酱 LeetCode 399. Evaluate Division

Solution 1: DFS

C++

Java

Python3

Solution 2: Union Find

C++

Java

Python3

花花酱 LeetCode 730. Count Different Palindromic Subsequences

Solution 1: Recursion with memoization

C++

Python

Java

Solution 2: DP

C++

Pyhton