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;
}
相关推荐
551只玄猫14 分钟前
【数学建模 matlab 实验报告12】聚类分析和判别分析
开发语言·数学建模·matlab·课程设计·聚类·实验报告
念恒123061 小时前
继承(下) (Inheritance)
c++
小陈工2 小时前
Python Web开发入门(十七):Vue.js与Python后端集成——让前后端真正“握手言和“
开发语言·前端·javascript·数据库·vue.js·人工智能·python
海清河晏1112 小时前
数据结构 | 单循环链表
数据结构·算法·链表
H Journey2 小时前
C++之 CMake、CMakeLists.txt、Makefile
开发语言·c++·makefile·cmake
wuweijianlove6 小时前
算法性能的渐近与非渐近行为对比的技术4
算法
研究点啥好呢6 小时前
Github热门项目推荐 | 创建你的像素风格!
c++·python·node.js·github·开源软件
_dindong6 小时前
cf1091div2 C.Grid Covering(数论)
c++·算法
AI成长日志6 小时前
【Agentic RL】1.1 什么是Agentic RL:从传统RL到智能体学习
人工智能·学习·算法
lly2024067 小时前
C 标准库 - `<stdio.h>`
开发语言