下面的程序有错误，请修改。

#include <iostream.h>

class A

{private:

int a;

public:

void func(B &);

A(int i){a=i;}

};

class B

{private:

int b;

friend void A::func(B &);

public:

B(int i){b=i;}

};

void A::func(B& r)

{a=r.b;

cout<<a<<endl;

}

void main()

{ B bt(3);

A at(10);

at.func(bt);

}

答案：［修改］class B;

class A［解析］class A类A中使用B类中的成员增加对B声明。