leetcode1146--快照数组

1. 题意

对一个数组进行多次修改，然后照一个快照。

再对数组进行询问第k次快照时的index处的值。

2. 题解

• 哈希表 + 二分
  对于每个位置，我们只需要顺序记录这个位置更改的快照和时间和值即可；
  再通过二分查找来获取它的值。

• 我的代码

class SnapshotArray {
private:
// index snap_cnt  val 
unordered_map<int, vector< pair<int,int>> > mem;

int snap_cnt;
public:
    SnapshotArray(int length):snap_cnt(0){

    }
    
    void set(int index, int val) {
        if ( mem.find( index ) == mem.end() ) {
            vector< pair<int,int>> tmp;
            mem[ index ] = tmp;   
        }

        if ( mem[ index ].empty()) {
            mem[ index ].push_back({snap_cnt, val});
        }
        else {
            int sz = mem[ index ].size();
            if ( mem[index][sz - 1].first != snap_cnt) {
                mem[ index ].push_back({snap_cnt, val});
            }
            else {
                mem[index][sz - 1].second = val;
            }
        }
    }
    
    int snap() {
        snap_cnt++;
        return snap_cnt - 1;
    }
    
    int get(int index, int snap_id) {
        if ( mem.find( index ) == mem.end() )
            return 0;   

        pair<int,int> p(snap_id, 0);

        auto low = lower_bound( mem[index].begin(), mem[index].end(), p, 
        [](const pair<int,int> &a, const pair<int,int> &b){
            return a.first < b.first;
        } );
        
        int sz = mem[ index ].size();
        if ( low == mem[index].end())
            return mem[ index ][ sz - 1 ].second;


        if ( low == mem[ index ].begin() && snap_id < low->first)
            return 0;

        if ( low->first == snap_id)
            return low->second;
        else {
            return (low - 1)->second;
        }
        
        return 0;
    }
};

/**
 * Your SnapshotArray object will be instantiated and called as such:
 * SnapshotArray* obj = new SnapshotArray(length);
 * obj->set(index,val);
 * int param_2 = obj->snap();
 * int param_3 = obj->get(index,snap_id);
 */

• 使用upper_bound

class SnapshotArray {
private:
    // index snap_cnt  val
    unordered_map<int, vector<pair<int, int>>> mem;

    int snap_cnt;

public:
    SnapshotArray(int length) : snap_cnt(0) {}

    void set(int index, int val) {
        // if ( mem.find( index ) == mem.end() ) {
        //     vector< pair<int,int>> tmp;
        //     mem[ index ] = tmp;
        // }

        // if ( mem[ index ].empty()) {
        //     mem[ index ].push_back({snap_cnt, val});
        // }
        // else {
        //     int sz = mem[ index ].size();
        //     if ( mem[index][sz - 1].first != snap_cnt) {
        //         mem[ index ].push_back({snap_cnt, val});
        //     }
        //     else {
        //         mem[index][sz - 1].second = val;
        //     }
        // }
        mem[index].emplace_back(snap_cnt, val);
    }

    int snap() {
        snap_cnt++;
        return snap_cnt - 1;
    }

    int get(int index, int snap_id) {
        pair<int, int> p(snap_id, 0);

        auto low =
            upper_bound(mem[index].begin(), mem[index].end(), p,
                        [](const pair<int, int>& a, const pair<int, int>& b) {
                            return a.first < b.first;
                        });

        return low == mem[index].begin() ? 0 : (low - 1)->second;
    }
};

/**
 * Your SnapshotArray object will be instantiated and called as such:
 * SnapshotArray* obj = new SnapshotArray(length);
 * obj->set(index,val);
 * int param_2 = obj->snap();
 * int param_3 = obj->get(index,snap_id);
 */