Class 02 递归回溯、前缀和、差分

递归回溯

给你一个长度为3的环形数组，请你往里面填数字1–20，要求不能重复，而且相邻两个数的和为质数。请输出所有的可能方案。

暴力法

#include<iostream>
#include<cstring>
using namespace std;


bool marked[21]; // 标记数组
int ans = 0;

// 判断是否为素数 
bool isPrime(int x){
	for(int i = 2; i * i <= x; i++){
		if(x % i == 0) return false;
	}
	return true;
} 


int main(){
	
	int arr[3];
	for(int i=1; i<=20; i++){
		if(marked[i]) continue;
		marked[i] = 1;
		arr[0] = i;
		
		for(int j=1; j<= 20; j++){
			if(marked[j]) continue;
			if(!isPrime(arr[0] + j)) continue;
			marked[j] = 1;
			arr[1] = j;
			
			for(int k=1; k<=20;k++){
				if(marked[k]) continue;
				if(!isPrime(arr[0] + k)) continue;
				if(!isPrime(arr[1] + k)) continue;
				ans++;
			}
			marked[j] = 0;
		}
		marked[i] = 0;
	}	

	cout<<ans<<endl;

	return 0;
}

回溯法

#include<iostream>
#include<cstring>
using namespace std;


bool marked[21]; // 标记数组
int ans = 0;
int a[100], n;

// 判断是否为素数 
bool isPrime(int x){
	for(int i = 2; i * i <= x; i++){
		if(x % i == 0) return false;
	}
	return true;
} 


void dfs(int level){
	if(level >= n + 1){
		if(isPrime(a[n] + a[1])) ans++;
		return;
	}
	
	for(int i=1; i <= 20; i++){
		if(!marked[i]){
			if(level > 1)
				if(!isPrime(i + a[level - 1])) continue;
			marked[i] = 1;
			a[level] = i;
			dfs(level + 1);
			marked[i] = 0;
		}
	}
}


int main(){
	
	n = 3;
	dfs(1);
	
	cout<<ans<<endl;

	return 0;
}

前缀和

在一个序列中，寻找下标 $l$ 到 下标 $r$ 之间的元素和
有大量询问，要求每次询问复杂度都是 $O(1)$

构造
$$S_i=a_1+a_2+\cdots +a_i$$
则
$$a_l+a_{l+1}+\cdots +a_r=S_r-S_{l-1}$$

支持逆运算的运算都可以前缀和
例如异或xor

for(int i=1;i<n;i++) 
	S[i] = S[i - 1] + a[i]; 
cout<<S[r] - S[l - 1]<<endl;

差分

区间加问题
将序列a的下标 $l$ 到 下标 $r$ 之间的每个元素都加上k，要求复杂度为 $O(n)$

我们使用 p[i] 去表示序列a元素之间的差值 a[i] - a[i-1]
我们需要将 需要增加的地方升起来、结尾之后降下来即可。

int []p; // 数组元素之间的差值
int []arr; // 需要的生成的数组

// 修改升起来和降下去的地方
void add(int le, int ri, int k){
	p[le] += k;
	p[ri + 1] -= k; 
}

// 根据元素之间的差值迭代生成arr数组
void createArr(){
	for(int i = 1; i <= n; i++){
		arr[i] = arr[i - 1] + p[i];
	}
}

题目

拓展阅读

• Fishdeer - 前缀和与差分