#include <iostream>
#include <map>
#include <cstdlib>
<p>// 用于记录分配信息
struct AllocInfo {
size_t size;
const char* file;
int line;
};</p><p>std::map<void*, AllocInfo> g_allocations;</p><p>// 重载 new
void<em> operator new(size_t size, const char</em> file, int line) {
void* ptr = std::malloc(size);
if (ptr) {
g_allocations[ptr] = {size, file, line};
std::cout << "ALLOC: " << ptr << " size=" << size 
<< " @ " << file << ":" << line << "\n";
}
return ptr;
}</p><p>// 提供普通 new 的重载（无文件行号）
void* operator new(size_t size) {
return ::operator new(size, "unknown", 0);
}</p><p>// 重载 delete
void operator delete(void* ptr) noexcept {
if (ptr == nullptr) return;</p><pre class='brush:php;toolbar:false;'>auto it = g_allocations.find(ptr);
if (it != g_allocations.end()) {
    std::cout << "FREE: " << ptr << " size=" << it->second.size 
              << " @ " << it->second.file << ":" << it->second.line << "\n";
    g_allocations.erase(it);
} else {
    std::cout << "FREE: " << ptr << " (UNKNOWN OR DOUBLE FREE!)\n";
}

std::free(ptr);

}

#define DEBUG_NEW new(__FILE__, __LINE__)
// 使用时替换 new 为 DEBUG_NEW
#define new DEBUG_NEW

void check_leaks() {
    if (g_allocations.empty()) {
        std::cout << "No memory leaks detected.\n";
    } else {
        std::cout << "Memory leaks detected:\n";
        for (const auto& pair : g_allocations) {
            std::cout << "LEAK: " << pair.first 
                      << " size=" << pair.second.size
                      << " @ " << pair.second.file 
                      << ":" << pair.second.line << "\n";
        }
    }
}