Codeforces 819D Mister B and Astronomers 数论+exgcd

题意

有$n$个人在观察天空，第$i\bmod n+1$个人会在第$i$个人观察完后的第$x_{i+1}$个时刻观察一个时刻的天空，其中第一个人在第$0$个时刻观察天空。现在天空中有一颗星星，只知道其每隔$T$秒会闪烁一次。对于每一个人，询问在$[0,T)$中有多少个整数时刻满足若星星在这个时刻闪烁，则这个人是第一个观察到星星的人。
$n\le200000,T\le10^9$

分析

设$y_i=\sum_{j=2}^{i}x_i$，显然若$i$和$j$满足$i<j且y_i\equiv y_j\pmod T$，那么第$j$个人的答案显然是$0$。
设$S=y_n$，不难发现第$i$个人第一次观察的时刻在$y_i\bmod T$，第二次观察在$y_i\bmod T+S$，如此类推。
那么若第$i$个人能看到某个位置$p$，必然有$y_i\equiv p\pmod{gcd(S,T)}$。
设$d=gcd(S,T)$，我们可以按照模$d$的值把所有位置分成$d$类，显然每一类是互不影响的。
对于每一类，恰好包含了$\frac{T}{d}$个位置，且某个处于这一类中的人，每次会往后跳$\frac{S}{d}$步，现在要对于每个人求有多少个格子满足第一个到达的人是他。
不难发现答案就是从他的位置开始，跳到第一个作为某个人起点的位置时，所跳的步数。
那么我们可以对每一类的位置编号，假设是第$i$类，那么位置$i$编号为$0$，$i+\frac{S}{d}$编号为$1$，如此类推。
这时若我们知道每一个位置这一类的人的编号，就可以很容易求出答案了。
而求某个位置的编号，可以用exgcd来做，这样总的复杂度就是$O(nlogn)$。

代码

#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;
}