您的位置:首页 > 其它

编写支持对其分配的malloc和free函数,分配内存时,malloc函数返回的地址必须能被2的n次方整除

2014-11-20 22:00 288 查看
题目:《程序员面试金典(第5版)》P281

编写支持对其分配的malloc和free函数,分配内存时,malloc函数返回的地址必须能被2的n次方整除。

//required_bytes为需要新建的字节数,alignment为首位置需要整除的数值
//按照题目要求,alignment应该是2的n次方
void* aligned_malloc(size_t required_bytes, size_t alignment)
{
	size_t offset = alignment - 1 + sizeof(void*);
	void* p = (void*)malloc(required_bytes+offset);
	if (p == nullptr)
		return nullptr;
	void* res = (void*)(((size_t)(p)+offset)&~(alignment-1));//关键
	void** tmp = (void**)res;
	tmp[-1] = p;	
	return res;
}

void aligned_free(void* r)
{
	if (r != nullptr)
	{
		void** tmp = (void**)r;
		free(tmp[-1]);
		r = nullptr;
	}
	
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐