2012-04-03 82 views
0

我正在編寫一個程序,用n個節點和高度k計算所有二叉樹。每個節點有0或2個孩子。該程序的工作原理,但我想添加一些memoization,因爲答案總是相同的一些特定的n和k。C++多維數組struct

我可以創建一個多維數組對,但我現在已經有了我的有用結構。我怎麼能聲明和使用這個mem變量。我在這方面沒有找到一個好的答案。我理解指針,但我更喜歡沒有內存管理的方法。

這是來自USACO訓練計劃btw的練習。

#include <iostream> 
#include <fstream> 
#include <string> 
#include <cmath> 
using namespace std; 

struct state { 
    int down, half; 
    state(int d, int h) : down(d), half(h) {} 
    int valid() { 
     return down != -1 && half != -1; 
    } 
}; 

state mem[200][100]; 

state cnt(int n, int k) 
{ 
    if (mem[n][k].valid()) 
     return mem[n][k]; 
    if (n == 1) 
     return state(k == 1, k != 1); 
    if (n > pow(2, k) - 1) 
     return state(-1, -1); 

    state total(0, 0); 
    for (int i = 1; i < n - 1; ++i) { 
     state left = cnt(i, k - 1); 
     state right = cnt(n - i - 1, k - 1); 

     if (left.valid() && right.valid()) { 
      total.down += left.down * right.down + 
          left.down * right.half + 
          left.half * right.down; 
      total.half += left.half * right.half; 
     } 
    } 

    return mem[n][k] = state(total.down % 9901, total.half % 9901); 
} 

int main() 
{ 
    ofstream fout ("nocows.out"); 
    ifstream fin ("nocows.in"); 

    int n, k; 
    fin >> n >> k; 

    for (int i = 0; i <= n; ++i) 
     for (int j = 0; j <= k; ++j) 
      mem[i][j] = state(-1, -1); 

    cout << cnt(n, k).down << endl; 

    return 0; 
} 

回答

3

您可以使用向量vector

std::vector<std::vector<state> > mem; 

您可以動態地添加到它,不用擔心尺寸(但如果你大致知道尺寸,你可以預先分配它以避免調整大小),並且內存清理也是自動的 - 當向量超出範圍時,其組件也將被刪除。

您的代碼不起作用,因爲您沒有state的默認構造函數。

事情是,當你編寫state mem[200][100];編譯器會嘗試創建100 * 200的state對象,但它不能。要做到這一點,你需要一個默認構造函數state

struct state { 
    state() : down(0), half(0) {} //default constructor 
    int down, half; 
    state(int d, int h) : down(d), half(h) {} 
    int valid() { 
     return down != -1 && half != -1; 
    } 
}; 
+0

感謝您的快速反應。這是我所需要的,但是你或其他人可以解釋爲什麼數組聲明不起作用? – Jasper 2012-04-03 18:12:54

+0

@Jasper在我的編輯中解釋過。 – 2012-04-03 18:18:08