C++ 稀疏表

C++ 稀疏表的作用是解决静态区间的最值问题,每次询问的时间复杂度可以降到O(1)。

稀疏表框架代码,见下

cpp 复制代码
#include<iostream>
#include<vector>
using namespace std;

// 静态区间最值问题

template<typename T, typename Compare>
class SparseTable {
private:
	vector<T> m_org;
	vector<vector<int>> m_st;
	Compare m_cmp;
	void init();
public:
	SparseTable(const vector<T>& arr);
	int Query(int l, int r);

};

template<typename T, typename Compare>
void SparseTable<T, Compare>::init()
{
	int n = (int)m_org.size();
	int log2n = (int)log2(n) + 1;
	m_st.assign(log2n, vector<int>(n));
	//st[j][i]表示的是[i, i+2^j-1]这个区间中的最值(所在的下标)
    //st[0][i]表示的是[i, i]这个区间中的最值(所在的下标),那就是i
	for (int i = 0; i < n; ++i) {
		m_st[0][i] = i;
	}

	for (int j = 1; (1 << j) <= n; ++j) {
		for (int i = 0; i + (1 << j) - 1 < n; ++i) {
			// [i, i+(1 << j)-1]
			/*
			  1, st[i][j]的区间长度是2^j
			  2, 把它拆成两个长度为2^(j-1)的区间
			     2.1 一个区间是[i, i+2^(j-1)-1] => st[j-1][i]
				 2.2 一个区间是[i + 2^(j-1), i+2^j-1] => st[j-1][i+1<<(j-1)]
			*/
			int idx1 = m_st[j - 1][i];
			int idx2 = m_st[j - 1][i + (1 << (j - 1))];
			m_st[j][i] = m_cmp(m_org[idx1], m_org[idx2]) ? idx1 : idx2;

		}
	}

}

template<typename T, typename Compare>
SparseTable<T, Compare>::SparseTable(const vector<T>& arr)
{
	int n = (int)arr.size();
	m_org.assign(arr.begin(), arr.end());
	init();
}
/*
1、对于[l, r]这个区间,可以拆分成两个长度分别为2^k的区间;
   一个区间是以1作为起始的:[1, x];
   一个区间是以r作为结尾的:[y, r];
   并且,这两个区间的并集是[l, r],所以需要满足 x+1>= y
2、以1作为起始的区间长度为2的k次
       它表示的区间就是[1, 1+2^j - 1] => x = 1 + 2^k - 1
   以r结尾的区间长度为2的k次
       它表示的区间就是[r - 2^k + 1, r] => y = r - 2^k + 1
3、1 + 2^k - 1 + 1 >= r - 2^k + 1
   移项: 2^(k+1) >= r - l + 1
   取对数:k >= log2(r - l + 1) - 1
   所以k的值,为log2(r-1+1)取上整 在减一

*/
template<typename T, typename Compare>
int SparseTable<T, Compare>::Query(int l, int r)
{
	if (l == r) {
		return l;
	}
	int k = (int)ceil(log2(r - 1 + 1)) - 1;
	int idx1 = m_st[k][l];
	int idx2 = m_st[k][r - (1 << k) + 1];
	return m_cmp(m_org[idx1], m_org[idx2]) ? idx1 : idx2;

}

//快速读入一个整数
inline int read()
{
	int x = 0, f = 1; char ch = getchar();
	while (ch < '0' || ch>'9') { if (ch == '-') f = -1; ch = getchar(); }
	while (ch >= '0' && ch <= '9') { x = x * 10 + ch - 48; ch = getchar(); }
	return x * f;
}


int main() {
	int n, m;
	n = read();
	m = read();
	vector<int> arr(n);
	for (int i = 0; i < n; ++i) {
		arr[i] = read();
	}
	SparseTable<int, std::greater<int>> st(arr);
	while (m--) {
		int l = read();
		int r = read();
		--l, --r;
		printf("%d\n",arr[st.Query(l, r)] );

	}
	return 0;
}

代码练习1 对应蓝桥云课 区间最大值 代码见下

cpp 复制代码
#include<iostream>
#include<vector>
using namespace std;

// 静态区间最值问题

template<typename T, typename Compare>
class SparseTable {
private:
	vector<T> m_org;
	vector<vector<int>> m_st;
	Compare m_cmp;
	void init();
public:
	SparseTable(const vector<T>& arr);
	int Query(int l, int r);

};

template<typename T, typename Compare>
void SparseTable<T, Compare>::init()
{
	int n = (int)m_org.size();
	int log2n = (int)log2(n) + 1;
	m_st.assign(log2n, vector<int>(n));
	//st[j][i]表示的是[i, i+2^j-1]这个区间中的最值(所在的下标)
    //st[0][i]表示的是[i, i]这个区间中的最值(所在的下标),那就是i
	for (int i = 0; i < n; ++i) {
		m_st[0][i] = i;
	}

	for (int j = 1; (1 << j) <= n; ++j) {
		for (int i = 0; i + (1 << j) - 1 < n; ++i) {
			// [i, i+(1 << j)-1]
			/*
			  1, st[i][j]的区间长度是2^j
			  2, 把它拆成两个长度为2^(j-1)的区间
			     2.1 一个区间是[i, i+2^(j-1)-1] => st[j-1][i]
				 2.2 一个区间是[i + 2^(j-1), i+2^j-1] => st[j-1][i+1<<(j-1)]
			*/
			int idx1 = m_st[j - 1][i];
			int idx2 = m_st[j - 1][i + (1 << (j - 1))];
			m_st[j][i] = m_cmp(m_org[idx1], m_org[idx2]) ? idx1 : idx2;

		}
	}

}

template<typename T, typename Compare>
SparseTable<T, Compare>::SparseTable(const vector<T>& arr)
{
	int n = (int)arr.size();
	m_org.assign(arr.begin(), arr.end());
	init();
}
/*
1、对于[l, r]这个区间,可以拆分成两个长度分别为2^k的区间;
   一个区间是以1作为起始的:[1, x];
   一个区间是以r作为结尾的:[y, r];
   并且,这两个区间的并集是[l, r],所以需要满足 x+1>= y
2、以1作为起始的区间长度为2的k次
       它表示的区间就是[1, 1+2^j - 1] => x = 1 + 2^k - 1
   以r结尾的区间长度为2的k次
       它表示的区间就是[r - 2^k + 1, r] => y = r - 2^k + 1
3、1 + 2^k - 1 + 1 >= r - 2^k + 1
   移项: 2^(k+1) >= r - l + 1
   取对数:k >= log2(r - l + 1) - 1
   所以k的值,为log2(r-1+1)取上整 在减一

*/
template<typename T, typename Compare>
int SparseTable<T, Compare>::Query(int l, int r)
{
	if (l == r) {
		return l;
	}
	int k = (int)ceil(log2(r - l + 1)) - 1;
	int idx1 = m_st[k][l];
	int idx2 = m_st[k][r - (1 << k) + 1];
	return m_cmp(m_org[idx1], m_org[idx2]) ? idx1 : idx2;

}

//快速读入一个整数
inline int read()
{
	int x = 0, f = 1; char ch = getchar();
	while (ch < '0' || ch>'9') { if (ch == '-') f = -1; ch = getchar(); }
	while (ch >= '0' && ch <= '9') { x = x * 10 + ch - 48; ch = getchar(); }
	return x * f;
}


int main() {
	int n, m;
	n = read();
	m = read();
	vector<int> arr(n);
	for (int i = 0; i < n; ++i) {
		arr[i] = read();
	}
	SparseTable<int, std::greater<int>> st(arr);
	while (m--) {
		int l = read();
		int r = read();
		--l, --r;
		printf("%d\n",arr[st.Query(l, r)] );

	}
	return 0;
}

代码练习 2 对应蓝桥云课 附近最小 代码见下

cpp 复制代码
#include<iostream>
#include<vector>
#include<cmath>
using namespace std;

// 静态区间最值问题

template<typename T, typename Compare>
class SparseTable {
private:
	vector<T> m_org;
	vector<vector<int>> m_st;
	Compare m_cmp;
	void init();
public:
	SparseTable(const vector<T>& arr);
	int Query(int l, int r);

};

template<typename T, typename Compare>
void SparseTable<T, Compare>::init()
{
	int n = (int)m_org.size();
	int log2n = (int)log2(n) + 1;
	m_st.assign(log2n, vector<int>(n));
	//st[j][i]表示的是[i, i+2^j-1]这个区间中的最值(所在的下标)
    //st[0][i]表示的是[i, i]这个区间中的最值(所在的下标),那就是i
	for (int i = 0; i < n; ++i) {
		m_st[0][i] = i;
	}

	for (int j = 1; (1 << j) <= n; ++j) {
		for (int i = 0; i + (1 << j) - 1 < n; ++i) {
			// [i, i+(1 << j)-1]
			/*
			  1, st[i][j]的区间长度是2^j
			  2, 把它拆成两个长度为2^(j-1)的区间
			     2.1 一个区间是[i, i+2^(j-1)-1] => st[j-1][i]
				 2.2 一个区间是[i + 2^(j-1), i+2^j-1] => st[j-1][i+1<<(j-1)]
			*/
			int idx1 = m_st[j - 1][i];
			int idx2 = m_st[j - 1][i + (1 << (j - 1))];
			m_st[j][i] = m_cmp(m_org[idx1], m_org[idx2]) ? idx1 : idx2;

		}
	}

}

template<typename T, typename Compare>
SparseTable<T, Compare>::SparseTable(const vector<T>& arr)
{
	int n = (int)arr.size();
	m_org.assign(arr.begin(), arr.end());
	init();
}
/*
1、对于[l, r]这个区间,可以拆分成两个长度分别为2^k的区间;
   一个区间是以1作为起始的:[1, x];
   一个区间是以r作为结尾的:[y, r];
   并且,这两个区间的并集是[l, r],所以需要满足 x+1>= y
2、以1作为起始的区间长度为2的k次
       它表示的区间就是[1, 1+2^j - 1] => x = 1 + 2^k - 1
   以r结尾的区间长度为2的k次
       它表示的区间就是[r - 2^k + 1, r] => y = r - 2^k + 1
3、1 + 2^k - 1 + 1 >= r - 2^k + 1
   移项: 2^(k+1) >= r - l + 1
   取对数:k >= log2(r - l + 1) - 1
   所以k的值,为log2(r-1+1)取上整 在减一

*/
template<typename T, typename Compare>
int SparseTable<T, Compare>::Query(int l, int r)
{
	if (l == r) {
		return l;
	}
	int k = (int)ceil(log2(r - l + 1)) - 1;
	int idx1 = m_st[k][l];
	int idx2 = m_st[k][r - (1 << k) + 1];
	return m_cmp(m_org[idx1], m_org[idx2]) ? idx1 : idx2;

}

//快速读入一个整数
inline int read()
{
	int x = 0, f = 1; char ch = getchar();
	while (ch < '0' || ch>'9') { if (ch == '-') f = -1; ch = getchar(); }
	while (ch >= '0' && ch <= '9') { x = x * 10 + ch - 48; ch = getchar(); }
	return x * f;
}


int main() {
	int n = read();
	vector<int> arr(n);
	for (int i = 0; i < n; ++i) {
		arr[i] = read();
	}
	int k = read();
	SparseTable<int, std::less<int>> st(arr);
	for (int i = 0; i < n; ++i) {
		int l = i - k;
		int r = i + k;
		if (l < 0) l = 0;
		if (r >= n) r = n - 1;
		printf("%d ", arr[st.Query(l, r)]);
	}
	printf("\n");

	return 0;
}

代码练习 3 对应蓝桥云课 区间选数k 代码见下

cpp 复制代码
#include<iostream>
#include<vector>
#include<cmath>
using namespace std;

// 静态区间最值问题

template<typename T, typename Compare>
class SparseTable {
private:
	vector<T> m_org;
	vector<vector<int>> m_st;
	Compare m_cmp;
	void init();
public:
	SparseTable(const vector<T>& arr);
	int Query(int l, int r);

};

template<typename T, typename Compare>
void SparseTable<T, Compare>::init()
{
	int n = (int)m_org.size();
	int log2n = (int)log2(n) + 1;
	m_st.assign(log2n, vector<int>(n));
	//st[j][i]表示的是[i, i+2^j-1]这个区间中的最值(所在的下标)
    //st[0][i]表示的是[i, i]这个区间中的最值(所在的下标),那就是i
	for (int i = 0; i < n; ++i) {
		m_st[0][i] = i;
	}

	for (int j = 1; (1 << j) <= n; ++j) {
		for (int i = 0; i + (1 << j) - 1 < n; ++i) {
			// [i, i+(1 << j)-1]
			/*
			  1, st[i][j]的区间长度是2^j
			  2, 把它拆成两个长度为2^(j-1)的区间
			     2.1 一个区间是[i, i+2^(j-1)-1] => st[j-1][i]
				 2.2 一个区间是[i + 2^(j-1), i+2^j-1] => st[j-1][i+1<<(j-1)]
			*/
			int idx1 = m_st[j - 1][i];
			int idx2 = m_st[j - 1][i + (1 << (j - 1))];
			m_st[j][i] = m_cmp(m_org[idx1], m_org[idx2]) ? idx1 : idx2;

		}
	}

}

template<typename T, typename Compare>
SparseTable<T, Compare>::SparseTable(const vector<T>& arr)
{
	int n = (int)arr.size();
	m_org.assign(arr.begin(), arr.end());
	init();
}
/*
1、对于[l, r]这个区间,可以拆分成两个长度分别为2^k的区间;
   一个区间是以1作为起始的:[1, x];
   一个区间是以r作为结尾的:[y, r];
   并且,这两个区间的并集是[l, r],所以需要满足 x+1>= y
2、以1作为起始的区间长度为2的k次
       它表示的区间就是[1, 1+2^j - 1] => x = 1 + 2^k - 1
   以r结尾的区间长度为2的k次
       它表示的区间就是[r - 2^k + 1, r] => y = r - 2^k + 1
3、1 + 2^k - 1 + 1 >= r - 2^k + 1
   移项: 2^(k+1) >= r - l + 1
   取对数:k >= log2(r - l + 1) - 1
   所以k的值,为log2(r-1+1)取上整 在减一

*/
template<typename T, typename Compare>
int SparseTable<T, Compare>::Query(int l, int r)
{
	if (l == r) {
		return l;
	}
	int k = (int)ceil(log2(r - l + 1)) - 1;
	int idx1 = m_st[k][l];
	int idx2 = m_st[k][r - (1 << k) + 1];
	return m_cmp(m_org[idx1], m_org[idx2]) ? idx1 : idx2;

}

//快速读入一个整数
inline int read()
{
	int x = 0, f = 1; char ch = getchar();
	while (ch < '0' || ch>'9') { if (ch == '-') f = -1; ch = getchar(); }
	while (ch >= '0' && ch <= '9') { x = x * 10 + ch - 48; ch = getchar(); }
	return x * f;
}


int main() {
	int n, m;
	n = read();
	m = read();
	vector<int> arr(n);
	for (int i = 0; i < n; ++i) {
		arr[i] = read();
	}
	SparseTable<int, std::less<int>> stl(arr);
	SparseTable<int, std::greater<int>> stg(arr);
	long long minsum = 0, maxsum = 0;
	while (m--) {
		int l = read();
		int r = read();
		--l, --r;
		minsum += arr[stl.Query(l, r)];
		maxsum += arr[stg.Query(l, r)];

	}
	cout << minsum << ' ' << maxsum << endl;
	return 0;
}
相关推荐
FFZero12 分钟前
C++ 内存模型与Memory Order深度解析
c++
永远都不秃头的程序员(互关)4 分钟前
C++核心语法复盘:数据结构编程的底层基石
开发语言
资深web全栈开发7 分钟前
LeetCode 3573. 买卖股票的最佳时机 V - 动态规划解法详解
算法·leetcode·动态规划
leiming69 分钟前
C++ 01 函数模板
开发语言·c++·算法
大道随心9 分钟前
【QT】自动化设备控制界面搭建
开发语言·qt·自动化
Chen--Xing11 分钟前
LeetCode LCR 119.最长连续序列
c++·python·算法·leetcode·rust
xiaoye-duck11 分钟前
吃透C++类和对象(上):封装、实例化与 this 指针详解
c++
金枪不摆鳍13 分钟前
算法2-链表
数据结构·算法·链表
yeshihouhou14 分钟前
redis数据分片算法
redis·算法·哈希算法