关注
#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;
}
查看原帖
点赞 评论
相关推荐
查看11道真题和解析 点赞 评论 收藏
分享
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 你的实习产出是真实的还是包装的? #
61823次浏览 535人参与
# 机械笔面试考察这些知识点 #
17773次浏览 121人参与
# 商战,最累的是我们 #
31392次浏览 97人参与
# AI时代还有必要刷leetcode吗? #
45387次浏览 548人参与
# 分享一个让你热爱工作的瞬间 #
69699次浏览 504人参与
# 米哈游笔试 #
671054次浏览 1206人参与
# 毕业租房也有小确幸 #
173406次浏览 4555人参与
# 字节7000实习来了,你投了吗? #
42142次浏览 316人参与
# 想从事Agent应该学习哪些技术? #
13116次浏览 379人参与
# 机械制造秋招总结 #
117630次浏览 927人参与
# 大学生该如何认清当下的就业环境? #
165526次浏览 905人参与
# 国企/银行/研究所公司爆料 #
208988次浏览 917人参与
# 26届春招投递记录 #
4708次浏览 49人参与
# 有哪些公司在面试时考察AICoding? #
26153次浏览 427人参与
# 实习必须要去大厂吗? #
202112次浏览 1784人参与
# HR面都在聊什么? #
20634次浏览 230人参与
# 初创公司值得加入吗? #
44638次浏览 226人参与
# 春招什么时候投? #
52358次浏览 709人参与
# 你都在哪些场所面过试? #
80812次浏览 506人参与
# 你现在一天AI几次? #
21863次浏览 147人参与