关注
#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;
}
查看原帖
点赞 评论
相关推荐
查看17道真题和解析 点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# HR问:你期望的薪资是多少?如何回答 #
67144次浏览 653人参与
# 我和mentor的爱恨情仇 #
77735次浏览 431人参与
# “vivo”个offer #
41440次浏览 288人参与
# 打工人的精神状态 #
98436次浏览 1288人参与
# 被同事甩锅了怎么办 #
24080次浏览 100人参与
# 你见过哪些工贼行为 #
28126次浏览 141人参与
# 一人一个landing小技巧 #
125387次浏览 1456人参与
# 你的秋招第一场笔试是哪家 #
258533次浏览 2026人参与
# 满帮集团求职进展汇总 #
6569次浏览 68人参与
# 考研失败就一定是坏事吗? #
149526次浏览 1081人参与
# vivo工作体验 #
29123次浏览 125人参与
# 和mentor 1on1 都聊什么? #
2123次浏览 21人参与
# 哪一瞬间觉得自己长大了 #
39409次浏览 494人参与
# 上班后和你想的一样吗? #
87963次浏览 671人参与
# 校招生月薪1W算什么水平 #
42945次浏览 234人参与
# 实习吐槽大会 #
386565次浏览 2156人参与
# 如何准备秋招 #
64571次浏览 830人参与
# 你想留在一线还是回老家? #
62651次浏览 537人参与
# 工作后明白的那些道理 #
22863次浏览 225人参与
# 数字马力求职进展汇总 #
206194次浏览 1635人参与
# 我是面试官,请用一句话让我破防 #
28771次浏览 132人参与

