Codeforces 1253A Single Push

思路:

即用数组b每个位置减去数组a对应位置的值,是否只存在一段连续的非零且相同的部分,且该差值需要大于0;

代码:

#define IOS ios::sync_with_stdio(false)
#include<bits/stdc++.h>
using namespace std;
#define rp(i,n) for(int i=0;i<n;i++)
#define rpn(i,n) for(int i=1;i<=n;i++)
const int MAX_N=1e5+99;
int a[MAX_N];
int main(){
	IOS;
	int t;
	cin>>t;
	while(t--){
		int n,k=0;
		cin>>n;
		rpn(i,n) cin>>a[i];
		bool flag=false,ok=true;
		rpn(i,n){
			int b;
			cin>>b;
			if(!ok) continue;
			int dis=b-a[i];
			if(dis){
				if(dis<0||flag||(k&&dis!=k)){ok=false;continue;} //之前出现过/差值不相等/差值小于0 
				if(!k) k=dis; 
			}
			else if(!flag&&!dis&&k) flag=true;
		}
		puts(ok?"YES":"NO");
	}
	return 0;
}
原文地址:https://www.cnblogs.com/yuhan-blog/p/12308856.html