Unique Pointer Custom Deleter

#include <iostream>
#include <cstdint>
#include <memory>
using namespace std;

class DataBlock {
public:
    uint8_t* buf1;
    uint8_t* buf2;
    int size;

    DataBlock(int n) : size(n) {
        buf1 = new uint8_t[n];
        buf2 = new uint8_t[n];
    }

    void set(int index, uint8_t value) {
        buf1[index] = value;
        buf2[index] = value;
    }

    void print() const {
        for (int i = 0; i < size; i++) {
            cout << (int)buf1[i];
            if (i != size - 1) cout << " ";
        }
        cout << endl;
    }
};

// Factory function creates object dynamically
DataBlock* createDataBlock(int n) {
    return new DataBlock(n);
}

// Custom cleanup function &#8212; must be used
void cleanup(DataBlock* p) {
    delete[] p->buf1;
    delete[] p->buf2;
    delete p;
    cout << "Object cleaned" << endl;
}

int main() {
    int N;
    cin >> N;

    {
        // Call createDataBlock()
        // Store returned object in unique_ptr with cleanup()
        unique_ptr<DataBlock, void (*)(DataBlock*)> ptr (createDataBlock(N), &cleanup);
        
        for (int i = 0; i < N; i++) {
            int temp;
            cin >> temp;
            // Store value in object
            ptr->set(i, temp);
        }
        ptr->print();
        // Print object data
    }

    return 0;
}

Solving Approach

 

 

 

 

 

Upvote
Downvote
Loading...

Input

1 0

Expected Output

0 Object cleaned