首頁 >後端開發 >C++ >C++程式用來檢查兩個字母堆疊是否可以被清空

C++程式用來檢查兩個字母堆疊是否可以被清空

PHPz
PHPz轉載
2023-09-04 14:01:061142瀏覽

C++程式用來檢查兩個字母堆疊是否可以被清空

假設有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中文網其他相關文章!

陳述:
本文轉載於:tutorialspoint.com。如有侵權,請聯絡admin@php.cn刪除