引用计数是一种常见的内存管理技巧,它主要是通过记录每个对象被引用的次数来决定它是否需要被释放。C++中有两种常见的引用计数方法,分别是智能指针和手动引用计数。以下是两种方法的示例代码和比较:
智能指针示例代码:
#include
class MyClass {
public:
int myVal;
};
int main() {
std::shared_ptr ptr1(new MyClass{});
std::shared_ptr ptr2 = ptr1;
std::cout << ptr1.use_count() << std::endl; // 输出2,表示有两个指针在引用这个对象和
std::cout << ptr2.use_count() << std::endl; // 输出2,同上
ptr1.reset();
std::cout << ptr1.use_count() << std::endl; // 输出0,表示没有指针再引用这个对象了
std::cout << ptr2.use_count() << std::endl; // 输出1,因为指针ptr2还在引用这个对象
}
手动引用计数示例代码:
class MyClass {
public:
int myVal;
int refCount;
MyClass(int val) : myVal(val), refCount(0) {}
void AddRef() {
refCount++;
}
void Release() {
refCount--;
if (refCount == 0) {
delete this;
}
}
};
int main() {
MyClass* obj = new MyClass(10);
obj->AddRef();
MyClass* obj2 = obj;
obj2->AddRef();
std::cout << obj->refCount << std::endl; // 输出2,因为有两个指针在引用这个对象
std::cout << obj2->refCount << std::endl; // 输出2,同上
obj->Release();
std::cout << obj->refCount << std::endl; // 输出1,只有一个指针再引用这个对象
std::cout << obj2->refCount << std::endl; // 输出1,同上
obj2->Release();
std::cout << obj->refCount << std::endl; // 输出0,没有指针再引用这个对象了,所以需要手动释放内存
std::cout << obj2->refCount << std::endl; //