给定正整数序列,每个数都不一样,求最长的子段[l,r]
,
满足a[l] % m = a[l+1] % m ... = a[r] % m , m>=2
从这个式子观察得到一个性质,如果[l,r]区间满足条件,那么a[l+1]-a[l],a[l+2]-a[l+1]一定是m的倍数,即:差分数组一定是k1 * m,k2 * m…的形式,ki是整数,
这个性质等价为[L,R]区间对应的差分数组的GCD大于等于2 。
而且我们发现,符合条件的区间长度是具有单调性的,
我们可以二分这个长度,然后枚举+验证,验证需要求区间GCD,用ST表、线段树都行。
O(n logn)
#include<bits/stdc++.h>
using namespace std;
//#pragma GCC optimize(2)
#define ull unsigned long long
#define ll long long
#define pii pair<int, int>
const int maxn = 2e5 + 10;
const ll mod = 998244353;
const ll inf = (ll)4e16+5;
const int INF = 1e9 + 7;
const double pi = acos(-1.0);
ll inv(ll b){if(b==1)return 1;return(mod-mod/b)*inv(mod%b)%mod;}
inline ll read()
{
ll 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-'0';ch=getchar();}
return x*f;
}
ll st[maxn][20];
int lg[maxn];
int n;
ll a[maxn];
inline ll qry(int l,int r)
{
int k=lg[r-l+1]-1;
return __gcd(st[l][k],st[r-(1<<k)+1][k]);
}
inline bool check(int x)
{
//差分数组 长度为x的区间 gcd大于1
for(int i=1;i+x-1<=n;i++)
{
int j=i+x-1;
if(qry(i,j)>1ll)
{
return 1;
}
}
return 0;
}
int main()
{
for(int i=1;i<maxn;i++) lg[i]=lg[i-1]+(1<<lg[i-1] == i);
int t;cin>>t;
while(t--)
{
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%lld",a+i);
n--;//差分数组 d[i]=a[i]-a[i+1] 数组d的长度为n-1
for(int i=1;i<=n;i++)
{
st[i][0]=abs(a[i]-a[i+1]);
}
for(int j=1;j<=20;j++)
{
for(int i=1;i+(1<<(j-1)) <= n;i++)
{
st[i][j]=__gcd(st[i][j-1],st[i+(1<<(j-1))][j-1]);
}
}
int l=1,r=n;
int ans=0;
while(l <= r)
{
int mid=l+r>>1;
if(check(mid))
{
ans=mid;
l=mid+1;
}
else r=mid-1;
}
printf("%d\n",ans+1);
}
return 0;
}
当然,我们也可以用双指针,先固定r指针不动,如果l指针对应位置不满足条件,那么l右移,求出以r为右端点的区间中,最优解。
#include<bits/stdc++.h>
using namespace std;
//#pragma GCC optimize(2)
#define ull unsigned long long
#define ll long long
#define pii pair<int, int>
const int maxn = 2e5 + 10;
const ll mod = 998244353;
const ll inf = (ll)4e16+5;
const int INF = 1e9 + 7;
const double pi = acos(-1.0);
ll inv(ll b){if(b==1)return 1;return(mod-mod/b)*inv(mod%b)%mod;}
inline ll read()
{
ll 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-'0';ch=getchar();}
return x*f;
}
ll st[maxn][20];
int lg[maxn];
int n;
ll a[maxn];
inline ll qry(int l,int r)
{
int k=lg[r-l+1]-1;
return __gcd(st[l][k],st[r-(1<<k)+1][k]);
}
int main()
{
for(int i=1;i<maxn;i++) lg[i]=lg[i-1]+(1<<lg[i-1] == i);
int t;cin>>t;
while(t--)
{
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%lld",a+i);
n--;
for(int i=1;i<=n;i++)
{
st[i][0]=abs(a[i]-a[i+1]);
}
for(int j=1;j<=20;j++)
{
for(int i=1;i+(1<<(j-1)) <= n;i++)
{
st[i][j]=__gcd(st[i][j-1],st[i+(1<<(j-1))][j-1]);
}
}
int ans=0;
int l=1,r=1;
//差分数组 最长区间 满足gcd大于1 双指针 区间长度满足单调性
while(l<=n && r<=n)
{
while(l<=r && qry(l,r)==1) l++;
ans=max(ans,r-l+1);
r++;
}
printf("%d\n",ans+1);
}
return 0;
}