CodeForces - 1634F Fibonacci Additions （思维）

是羊驼鸭

已于 2022-03-19 19:33:45 修改

阅读量206

点赞数

分类专栏：思维、规律文章标签：算法

于 2022-03-19 19:33:38 首次发布

本文链接：https://blog.csdn.net/qq_39641976/article/details/123600665

版权

思维、规律专栏收录该内容

53 篇文章 0 订阅

订阅专栏

该博客主要讨论了一种基于Fibonacci数列的区间加法操作，以及如何在操作后判断两个数列在模意义下是否相等。博主通过引入差分数组和维护0元素个数的方法，实现了O(n+q)时间复杂度的解决方案，其中n是数列长度，q是操作次数。代码展示了如何处理区间加法并快速响应询问。

摘要由CSDN通过智能技术生成

题目链接：点击这里

题目大意：
已知两个长度相同的数组 $A$ ， $B$ ，给出若干次操作，每次操作后你需要求出取模过后的数组 $A$ ， $B$ 是否相等。

有两个长度都为 $n$ 的数列 $A, B $ ，同时他会进行 $q$ 次操作。

对于每一次操作，他会先选择其中一个数列 $(A / B)$ ，再选择一个区间 $[l, r] (1 \leq l \leq r \leq n)$ ，将选定的序列 $[l, r]$ 中的数对位加上 $\text{Fibonacci}$ 数列。

换句话说，就是将选定数列的第 $l$ 项加上 $1$ ，第 $l + 1$ 项加上 $1$ ，第 $l + 2$ 项加上 $2$ ，第 $l + 3$ 项加上 $3$ … 第 $r$ 项加上 $F i b (r - l + 1)$ ，即 $\text{Fibonacci}$ 数列的第 $r - l + 1$ 项。

在每次操作结束的时候，询问此时的 $A$ 和 $B$ 两个序列在模 M 的意义下是否相同

题目分析：
我们引入一个新数组 $C(c_i=a_i-b_i)$ ，若 $A = = B$ 则 $c_i=0$ 恒成立，所以我们可以通过 $C$ 数组来判断 $A, B$ 是否相等
那么如何实现区间加操作呢，我们用 $c_i$ 构造一个 $d_i=c_i-c_{i-1}-c_{i-2}$ 这样我们就可以使用类似于差分的技巧实现区间加了，只需要在区间的两个端点修改点值即可
若要区间 $[l, r]$ 加上 $\text{Fibonacci}$ 数列，只需要让 $d [l] + 1, d [r + 1] - F i b [r - l + 2], d [r + 2] = F i b [r - l + 1]$ 即可，这样每次修改操作的时间复杂度就变成 $O (1)$ 了
我们发现当且仅当 $c_i$ 恒等于 $0$ 时 $d_i$ 才会恒等于 $0$ ，所以我们可以维护 $d [i]$ 中 $0$ 的个数，如果全为 $0$ 则说明相等，不全为 $0$ 则说明不相等
整体时间复杂度为 $O (n + q)$

具体细节见代码：

// Problem: F. Fibonacci Additions
// Contest: Codeforces - Codeforces Round #770 (Div. 2)
// URL: https://codeforces.com/contest/1634/problem/F
// Memory Limit: 256 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

//#pragma GCC optimize(2)
//#pragma GCC optimize("Ofast","inline","-ffast-math")
//#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<set>
#include<map>
#include<stack>
#include<queue>
#include<unordered_map>
#define ll long long
#define inf 0x3f3f3f3f
#define Inf 0x3f3f3f3f3f3f3f3f
//#define int  ll
#define endl '\n'
#define IOS ios::sync_with_stdio(0); cin.tie(0); cout.tie(0)
using namespace std;
int read()
{
	int res = 0,flag = 1;
	char ch = getchar();
	while(ch<'0' || ch>'9')
	{
		if(ch == '-') flag = -1;
		ch = getchar();
	}
	while(ch>='0' && ch<='9')
	{
		res = (res<<3)+(res<<1)+(ch^48);//res*10+ch-'0';
		ch = getchar();
	}
	return res*flag;
}
const int maxn = 1e6+5;
// const int mod = 1e9+7;
const double pi = acos(-1);
const double eps = 1e-8;
int n,q,mod,a[maxn],b[maxn],c[maxn],d[maxn],cnt,f[maxn];
char s[5];
int main()
{
	n = read(),q = read(),mod = read();
	f[1] = f[2] = 1;
	for(int i = 3;i <= n;i++) f[i] = (f[i-1]+f[i-2])%mod;
	for(int i = 1;i <= n;i++) a[i] = read();
	for(int i = 1;i <= n;i++) b[i] = read();
	for(int i = 1;i <= n;i++) c[i] = ((a[i]-b[i])%mod+mod)%mod;
	for(int i = 1;i <= n;i++) d[i] = i==1 ? c[i] : (((c[i]-c[i-1])%mod-c[i-2])%mod+mod)%mod,cnt += d[i]==0;
	while(q--)
	{
		scanf("%s",s);
		int l = read(),r = read();
		if(s[0] == 'A')
		{
			if(d[l] == 0) cnt--;
			if(r+1<=n && d[r+1]==0) cnt--;
			if(r+2<=n && d[r+2]==0) cnt--;
			d[l] = (d[l]+1)%mod;
			d[r+1] = (d[r+1]-f[r-l+2]+mod)%mod;
			d[r+2] = (d[r+2]-f[r-l+1]%mod)%mod;
			if(d[l] == 0) cnt++;
			if(r+1<=n && d[r+1]==0) cnt++;
			if(r+2<=n && d[r+2]==0) cnt++;
		}
		else {
			if(d[l] == 0) cnt--;
			if(r+1<=n && d[r+1]==0) cnt--;
			if(r+2<=n && d[r+2]==0) cnt--;
			d[l] = (d[l]-1+mod)%mod;
			d[r+1] = (d[r+1]+f[r-l+2])%mod;
			d[r+2] = (d[r+2]+f[r-l+1])%mod;
			if(d[l] == 0) cnt++;
			if(r+1<=n && d[r+1]==0) cnt++;
			if(r+2<=n && d[r+2]==0) cnt++;
		}
		if(cnt == n) puts("YES");
		else puts("NO");
	}
	return 0;
}