假設有2n封信,每封信上都寫有1到n之間的整數。有兩封信上寫有相同的數字。這些信被分成m堆,第i堆上有stack[i]封信。我們的任務是以以下方式清空所有堆:
我們必須選擇任兩堆,並從兩堆中移除頂部的信件。
我們所移除的信件必須具有相同的數字。
如果我們能以這種方式清空m堆,則列印true,否則傳回false。
因此,如果輸入是n = 3,m = 2,stacks = {{2, 1, 3}, {2, 1, 3}},那麼輸出將為true。
有兩堆,每堆上的信件上分別寫有數字2、1、3。因此,我們可以按照給定的方式從兩堆中移除信件並清空它們。
為了解決這個問題,我們將按照以下步驟進行:
Define one 2D array dp Define an array tvec for initialize i := 0, when i < m, update (increase i by 1), do: k := size of stacks[i] for initialize j := 0, when j < k, update (increase j by 1), do: if j < 0, then: insert p at the end of dp[stacks[i, j]] (increase tvec[p] by 1) p := stacks[i, j] Define an array tp for initialize i := 1, when i <= n, update (increase i by 1), do: Define one queue q insert i into q while (q is not empty), do: if not tp[first element of q] and tvec[first element of q] is same as 0, then: for each element next in dp[first element of q], do: (decrease tvec[next] by 1) insert next into q tp[first element of q] := true delete first element from q for initialize i := 1, when i <= n, update (increase i by 1), do: if tvec[i] is not equal to 0, then: return false return true
讓我們看看以下實現,以便更好地理解-
#include <bits/stdc++.h> using namespace std; bool solve(int n, int m, vector<vector<int>> stacks){ vector<vector<int>> dp(n + 1); vector<int> tvec(n + 1); for(int i = 0; i < m; i++){ int k = stacks[i].size(); int p; for(int j = 0; j < k; j++){ if(j > 0){ dp[stacks[i][j]].push_back(p); tvec[p]++; } p = stacks[i][j]; } } vector<bool> tp(n + 1); for(int i = 1; i <= n; i++){ queue<int> q; q.push(i); while(!q.empty()){ if(!tp[q.front()] && tvec[q.front()] == 0){ for(auto next: dp[q.front()]){ tvec[next]--; q.push(next); } tp[q.front()]=true; } q.pop(); } } for(int i = 1; i <= n; i++){ if(tvec[i] != 0){ return false; } } return true; } int main() { int n = 3, m = 2; vector<vector<int>> stacks = {{2, 1, 3}, {2, 1, 3}}; cout<< solve(n, m, stacks); return 0; }
3, 2, {{2, 1, 3}, {2, 1, 3}}
1
以上是C++程式用來檢查兩個字母堆疊是否可以被清空的詳細內容。更多資訊請關注PHP中文網其他相關文章!