静的 RMQ(更新のない区間最小値クエリ) に O(1) で答えるデータ構造を O(nlogn) 時間で構築するアルゴリズム.
「table[k][i] : インデックス i から始まる 2k 個の要素の中の最小値」 という配列を O(nlogn) かけて前計算しておけば区間 [l,r) のクエリに対して
k=⌊log2(r−l)⌋ とすると min(table[k][i],table[k][r−2k]) のようにして O(1) で答えることができる.
ここでテーブルの添字は前計算の際にキャッシュに乗りやすいように k, i の順の方が良い.
ちなみに和や xor などもう少し広いクラス(半群をなすもの) に対してもクエリの計算時間 O(1) を達成するデータ構造を考えることができ,
本アルゴリズムと異なりクエリで呼び出される区間が重ならないことから disjoint sparse table と呼ばれている(参考 noshi さんのブログ).
時間計算量: 構築 O(nlogn), クエリ O(1)
- template<typename T> class SparseTable {
- private:
- int sz;
- vector<int> LogTable;
- vector<vector<T> > Table; //最小値を保持
- public:
- SparseTable(const vector<T>& v) : sz((int)v.size()), LogTable(sz + 1){
- for(int i = 2; i < sz + 1; i++){
- LogTable[i] = LogTable[i >> 1] + 1;
- }
- Table.resize(LogTable[sz]+1, vector<T>(sz));
- for(int i = 0; i < sz; i++){
- Table[0][i] = v[i];
- }
- for(int k = 1; (1 << k) <= sz; k++){
- for(int i = 0; i + (1 << k) <= sz; i++){
- Table[k][i] = min(Table[k-1][i], Table[k-1][i + (1 << (k-1))]);
- }
- }
- }
- T query(const int l, const int r){
- const int k = LogTable[r-l];
- return min(Table[k][l], Table[k][r-(1<<k)]);
- }
- };
Atcooder : Yakiniku Restaurants 提出コード