c++ - 关于模板类的友元

Question

先上代码 :

#ifndef AVLTREE_H
#define AVLTREE_H

template <typename T>
class AvlTree {
private:
    AvlTree* leftChild;
    AvlTree* rightChild;
    int height;
    T* element;

public:
    AvlTree() :leftChild(0), rightChild(0), height(0), element(0) {}
    AvlTree(T& item) :AvlTree() {
        element = new T(item);
    }
    ~AvlTree() {
        delete element;
    }

private:
    friend int getHeight(AvlTree<T>*);
};

template <typename T>
int getHeight(AvlTree<T>* tree) {
    return tree ? tree->height : -1;
}
#endif // !AVLTREE_H

#include <iostream>
#include <string>
#include "test.h"

int main(void) {
    AvlTree<int>* x = NULL;
    std::cout << getHeight(x) << std::endl;
}

报错信息 :

[zhangzhimin@ c++] $ g++ main.cpp --std=c++11
Undefined symbols for architecture x86_64:
  "getHeight(AvlTree<int>*)", referenced from:
      _main in main-fcb2cb.o
ld: symbol(s) not found for architecture x86_64
clang: error: linker command failed with exit code 1 (use -v to see invocation)
[zhangzhimin@ c++] $ 

我是想让 getHeight 这个函数成为模板类AvlTree的友元, 要那种一对一的友元, 就比如AvlTree<int>它只和getHeight(AvlTree<int>)成为友元, 不知道这上面的代码为什么错了?

Answer 1

類別裡面宣告的是一個普通函數，但是類別外面定義的是一個模板函數。
可以直接在類別裡面定義那個函數，註解掉外面的模板函數。類別模板會給你產生一個getHeight.

    friend int getHeight(AvlTree<T>* tree){
        return tree ? tree->height : -1;
    }

或

private:
    template <typename U>
    friend int getHeight(AvlTree<U>* tree);//{
        //return tree ? tree->height : -1;
   // }
};

template <typename T>
int getHeight(AvlTree<T>* tree) {
     return tree ? tree->height : -1;
}

Answer 2

那應該是這樣子

private:
    template <typename X>
    friend int getHeight(AvlTree<X>*);

Answer 3

記得 一書中有一節專門講這個問題