Description
小H是个善于思考的学生,现在她又在思考一个有关序列的问题。
她的面前浮现出一个长度为n的序列{ai},她想找出一段区间[L, R](1 <= L <= R <= n)。
这个特殊区间满足,存在一个k(L <= k <= R),并且对于任意的i(L <= i <= R),ai都能被ak整除。这样的一个特殊区间 [L, R]价值为R - L。
小H想知道序列中所有特殊区间的最大价值是多少,而有多少个这样的区间呢?这些区间又分别是哪些呢?你能帮助她吧。
Input
第一行,一个整数n.
第二行,n个整数,代表ai.
Output
第一行两个整数,num和val,表示价值最大的特殊区间的个数以及最大价值。
第二行num个整数,按升序输出每个价值最大的特殊区间的L.
Sample Input
输入1:
5
4 6 9 3 6
输入2:
5
2 3 5 7 11
Sample Output
输出1:
1 3
2
输出2:
5 0
1 2 3 4 5
Data Constraint
30%: 1 <= n <= 30 , 1 <= ai <= 32.
60%: 1 <= n <= 3000 , 1 <= ai <= 1024.
这里是引用
80%: 1 <= n <= 300000 , 1 <= ai <= 1048576.
100%: 1 <= n <= 500000 , 1 <= ai < 2 ^ 31.
这道题,我立马想到二分答案,因为长度符合单调性。
二分答案,另一个重点就是判断答案对错,显然暴力判答案也是会错的。
所以我想到了维护区间最小值与最大公约数,RMQ(ST算法)
首先我们知道最小值是可以用ST维护的,
但为什么可以用ST维护最大公约数呢?
因为我们知道已知
那么
所以重叠也无所谓。
算法步骤:
二分出值mid
枚举起点i
算出i~i+x-1符不符合要求(即最小值是否为最大公约数)
代码:
#include<iostream>
#include<cstdio>
using namespace std;
int n,i,j,l,r,mid,anstot,an;
int a[500001],f[500001][20],ff[500001][20],log[500001],ans[500001],anss[500001];
int gcd(int x,int y)
{
if (x%y==0) return y;
return gcd(y,x%y);
}
int check(int x)
{
int reply=0,t,tt,tot=0,k,bb;
for (k=1;k<=n-x+1;k++)
{
bb=0;
t=gcd(f[k][log[x]],f[(k+x-1)-(1<<log[x])+1][log[x]]);
tt=min(ff[k][log[x]],ff[(k+x-1)-(1<<log[x])+1][log[x]]);
if (t==tt)
{
reply=1;
tot++;
ans[tot]=k;
}
}
if (reply)
{
anstot=tot;
for (k=1;k<=tot;k++)
anss[k]=ans[k];
}
return reply;
}
int main()
{
scanf("%d",&n);
log[0]=-1;
for (i=1;i<=n;i++)
{
scanf("%d",&a[i]);
f[i][0]=a[i];
ff[i][0]=a[i];
log[i]=log[i/2]+1;
}
for (j=1;j<=log[n];j++)
for (i=1;i+(1<<j)-1<=n;i++)
{
f[i][j]=gcd(f[i][j-1],f[i+(1<<j-1)][j-1]);
ff[i][j]=min(ff[i][j-1],ff[i+(1<<j-1)][j-1]);
}
l=0;
r=n-1;
while (l<=r)
{
mid=(l+r)/2;
if (check(mid+1))
{
an=mid;
l=mid+1;
}
else r=mid-1;
}
printf("%d %d\n",anstot,an);
for (i=1;i<=anstot;i++) printf("%d ",anss[i]);
}
Tips:
RMQ具体操作详见楼主博客,“浅谈RMQ”