关注
#include <iostream>
using namespace std;
template<typename T>
class SharedPtr
{
public:
//禁止分配内存时抛出异常
explicit SharedPtr(T *p):ptr(p), use(new (nothrow) size_t(1))
{
//分配内存失败
if (use == nullptr)
{
delete ptr;
ptr = nullptr;
cout << "分配内存失败";
exit(-1);
}
}
explicit SharedPtr(const SharedPtr &rhs) :ptr(rhs.ptr), use(rhs.use)
{
++*use;
}
SharedPtr& operator=(const SharedPtr &rhs)
{
++*rhs.use;
if (--*use == 0)
{
delete ptr;
ptr = nullptr;
delete use;
use = nullptr;
}
ptr = rhs.ptr;
use = rhs.use;
return *this;
}
T* get() const
{
return ptr;
}
size_t use_count() const
{
return *use;
}
bool unique() const
{
return use_count() == 1;
}
//交换两个指针的指向
void swap(SharedPtr &q)
{
SharedPtr temp(q);
q = *this;
*this = temp;
}
T& operator*() const
{
return *ptr;
}
T* operator->() const
{
return & operator*();
}
//虚析构函数
virtual ~SharedPtr()
{
if (--*use == 0)
{
delete ptr;
ptr = nullptr;
delete use;
use = nullptr;
}
}
private:
T *ptr;
size_t *use;
};
int main()
{
//测试
SharedPtr<int> s1(new int(666));
{
SharedPtr<int> s2(s1);
SharedPtr<int> s3(s2);
cout << "*s1 = " << *s1 << endl;
cout << "s1 use_count " << s1.use_count() << endl;
cout << "s1 is unique? " << s1.unique() << endl;
}
SharedPtr<int> s4(new int(888));
SharedPtr<int> s5(s4);
cout << "Before swap:\n";
cout << "*s5 " << *s5 << endl;
cout << "*s1 " << *s1 << endl;
s5.swap(s1);
cout << "After swap:\n";
cout << "*s5 " << *s5 << endl;
cout << "*s1 " << *s1 << endl;
cout << *(s1.get()) << endl;
s1 = s4;
SharedPtr<double> pd1(new double(2.3333));
SharedPtr<double> pd2(new double(0.33333));
cout << "pd1 is unique? " << pd1.unique() << endl;
auto q = &pd1;
cout << "Before swap:\n";
cout << "*pd1 " << *pd1 << endl;
cout << "*pd2 " << *pd2 << endl;
q->swap(pd2);
cout << "After swap:\n";
cout << "*pd1 " << *pd1 << endl;
cout << "*pd2 " << *pd2 << endl;
cout << "test end!\n";
return 0;
}
查看原帖
点赞 评论
相关推荐
牛客热帖
更多
正在热议
更多
# 帮我看看,领导说这话什么意思? #
15584次浏览 87人参与
# 牛友的志愿填报指南 #
33495次浏览 179人参与
# 快手技术岗信息交流阵地 #
127次浏览 0人参与
# 你的mentor是什么样的人? #
12366次浏览 97人参与
# 毕业租房也有小确幸 #
140239次浏览 4491人参与
# 怎么给家人解释你的工作? #
7534次浏览 57人参与
# 得物app工作体验 #
27132次浏览 61人参与
# 租房前辈的忠告 #
259122次浏览 7114人参与
# 国企还是互联网,你怎么选? #
167561次浏览 1191人参与
# 求职中的尴尬瞬间 #
1608次浏览 24人参与
# 小红书求职进展汇总 #
120199次浏览 951人参与
# 薪资爆料 #
199588次浏览 1511人参与
# 校招泡的最久的公司是哪家? #
10143次浏览 65人参与
# 求职低谷期你是怎么度过的 #
10103次浏览 199人参与
# 26届秋招公司红黑榜 #
24088次浏览 87人参与
# 从哪些方向判断这个offer值不值得去? #
12751次浏览 157人参与
# 度小满求职进展汇总 #
11897次浏览 64人参与
# 你觉得mentor喜欢什么样的实习生 #
14810次浏览 392人参与
# 牛客树洞,我想对你说 #
4233次浏览 62人参与
# 还记得你第一次面试吗? #
339945次浏览 3876人参与
# 机械人的秋招小目标 #
22586次浏览 217人参与