题意
有
n
n
个人在观察天空,第个人会在第
i
i
个人观察完后的第个时刻观察一个时刻的天空,其中第一个人在第
0
0
个时刻观察天空。现在天空中有一颗星星,只知道其每隔秒会闪烁一次。对于每一个人,询问在
[0,T)
[
0
,
T
)
中有多少个整数时刻满足若星星在这个时刻闪烁,则这个人是第一个观察到星星的人。
n≤200000,T≤109
n
≤
200000
,
T
≤
10
9
分析
设
yi=∑ij=2xi
y
i
=
∑
j
=
2
i
x
i
,显然若
i
i
和满足
i<j且yi≡yj(modT)
i
<
j
且
y
i
≡
y
j
(
mod
T
)
,那么第
j
j
个人的答案显然是。
设
S=yn
S
=
y
n
,不难发现第
i
i
个人第一次观察的时刻在,第二次观察在
yimodT+S
y
i
mod
T
+
S
,如此类推。
那么若第
i
i
个人能看到某个位置,必然有
yi≡p(modgcd(S,T))
y
i
≡
p
(
mod
g
c
d
(
S
,
T
)
)
。
设
d=gcd(S,T)
d
=
g
c
d
(
S
,
T
)
,我们可以按照模
d
d
的值把所有位置分成类,显然每一类是互不影响的。
对于每一类,恰好包含了
Td
T
d
个位置,且某个处于这一类中的人,每次会往后跳
Sd
S
d
步,现在要对于每个人求有多少个格子满足第一个到达的人是他。
不难发现答案就是从他的位置开始,跳到第一个作为某个人起点的位置时,所跳的步数。
那么我们可以对每一类的位置编号,假设是第
i
i
类,那么位置编号为
0
0
,编号为
1
1
,如此类推。
这时若我们知道每一个位置这一类的人的编号,就可以很容易求出答案了。
而求某个位置的编号,可以用exgcd来做,这样总的复杂度就是。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<map>
#include<set>
typedef long long LL;
const int N=200005;
int n,T,S,y[N],pos[N],sz;
std::map<int,int> ma,id;
std::set<int> se[N];
std::set<int>::iterator it;
bool vis[N];
int gcd(int x,int y)
{
if (!y) return x;
else return gcd(y,x%y);
}
void exgcd(int x,int y,int &a,int &b)
{
if (!y) {a=1;b=0;return;}
exgcd(y,x%y,a,b);
int tmp=a;a=b;b=tmp-(x/y)*b;
}
int main()
{
scanf("%d%d",&T,&n);
for (int i=1;i<=n;i++)
{
int x;scanf("%d",&x);
if (i>1) y[i]=(y[i-1]+x)%T;
(S+=x)%=T;
if (!ma[y[i]]) ma[y[i]]=1,vis[i]=1;
}
int a,b,d=gcd(S,T);
S/=d;T/=d;
exgcd(S,T,a,b);
for (int i=1;i<=n;i++)
{
int z=y[i]%d;
if (!id[z]) id[z]=++sz;
pos[i]=(LL)a*(y[i]-z)/d%T;
se[id[z]].insert(pos[i]);
}
for (int i=1;i<=n;i++)
{
if (!vis[i]) {printf("%d ",0);continue;}
int z=id[y[i]%d];
it=se[z].lower_bound(pos[i]);
it++;
if (it!=se[z].end()) printf("%d ",(*it)-pos[i]);
else printf("%d ",(*se[z].begin())+T-pos[i]);
}
return 0;
}