week1 星期二

差分：

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

const int N = 200010;

int b[N];

int main() {
	int T;
	cin >> T;
	while (T--) {
		int n;
		cin >> n;
		memset(b,0,(n+1)*4);
		for (int i = 1; i <= n; i++) {
		    
			int x;
			cin>>x;
			x = min(x,i);
			int l = i - x + 1,r = i;
			b[l]++;b[r+1]--;
		}
		for(int i = 1;i <=n;i++) b[i] += b[i-1];
		for (int i = 1; i <= n; i++) printf("%d ",!!b[i]);
		puts("");
	}
	return 0;
}

本题用到差分的思想解决，一开始用暴力的做法做会超时，如下：

#include<iostream>
using namespace std;

const int N = 200010;

int a[N];

int main() {
	int T;
	cin >> T;
	while (T--) {
		int n;
		cin >> n;
		for (int i = 0; i < n; i++) {
			int k;
			cin >> k;
			a[i] = 0;
			for (int j = i; j > i - k && j >= 0; j--) {
				a[j] = 1;
			}
		}
		for (int i = 0; i < n; i++) {
			cout << a[i] << " ";
		}
		cout << endl;
	}
	return 0;
}

复习一下差分，相当于前缀和的逆操作，a作为前缀数组，b作为原数组，a[i]为b数组前i个元素的和，那么a[i]-a[l-1]相当于b数组在区间[l,r]上的和，那么计算和就可以从O(N)时间复杂度转换为O（1），差分则a[l,r]区间上加上一个数c，那么只要将b数组b[l]+c、b[r+1]-c就可以实现，从l开始，到r这个区间内，前缀和都加上数c，但当r+1时，前缀和保持不变，那么就减去c。

将要让a[l,r]变成一个数k也可以用这样的方法来进行，转换成将a[l,r]区间都加上一个数，如果没有被加到的区间就保持原样，如果加到的就变成k，判断一下即可。