AT2005 [AGC003E] Sequential operations on Sequence

题目链接:AGC003E

给定一个初始为 $1\sim n$ 的序列,给定 $q$ 次操作,每次操作把数组长度变为 $q_i$,新增/减少的数为上一个操作后的数组的重复。问 $q$ 次操作后,$1\sim n$ 每个数出现次数。

$n,q \leq 10^5,q_i\leq 10^{18}$。

Tutorial

容易观察得到一个小性质:如果 $q_i > q_{i+1}$,则 $q_i$ 操作一定是无效的。

于是现在操作序列变为递增的。

考虑一次操作一定是把所有 $q_i$ 对应的贡献先乘上 $\lfloor \frac{q_i}{q_{i-1}} \rfloor$,再处理 $q_i \bmod q_{i-1}$ 部分的边角。

设 $f_i$ 表示该次序列对最终序列的贡献,于是最后倒推一遍即可。

注意到边角部分一定会出现在 $q_j < q_i \bmod q_{i-1},q_i \bmod q_{i-1} > q_{j+1}$,于是将拆出的 $q_j$ 数量加入贡献,剩余部分继续拆分。

由于每次拆一定会 $/2$,于是差分一下,复杂度是 $O(N\log ^2 N)$。

Solution

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#include<bits/stdc++.h>
#define Tp template<typename Ty>
#define Ts template<typename Ty,typename... Ar>
#define W while
#define I inline
#define RI register int
#define LL long long
#define Cn const
#define CI Cn int&
#define gc getchar
#define D isdigit(c=gc())
#define pc(c) putchar((c))
using namespace std;
namespace Debug{
Tp I void _debug(Cn char* f,Ty t){cerr<<f<<'='<<t<<endl;}
Ts I void _debug(Cn char* f,Ty x,Ar... y){W(*f!=',') cerr<<*f++;cerr<<'='<<x<<",";_debug(f+1,y...);}
Tp ostream& operator<<(ostream& os,Cn vector<Ty>& V){os<<"[";for(Cn auto& vv:V) os<<vv<<",";os<<"]";return os;}
#define gdb(...) _debug(#__VA_ARGS__,__VA_ARGS__)
}using namespace Debug;
namespace FastIO{
Tp I void read(Ty& x){char c;int f=1;x=0;W(!D) f=c^'-'?1:-1;W(x=(x<<3)+(x<<1)+(c&15),D);x*=f;}
Ts I void read(Ty& x,Ar&... y){read(x),read(y...);}
Tp I void write(Ty x){x<0&&(pc('-'),x=-x,0),x<10?(pc(x+'0'),0):(write(x/10),pc(x%10+'0'),0);}
Tp I void writeln(Cn Ty& x){write(x),pc('\n');}
}using namespace FastIO;
Cn int N=1e5+10;
int n,q;LL a[N],f[N],g[N];
I void Sol(LL d,LL w){
RI j=upper_bound(a+1,a+n+1,d)-a-1;
if(!j) g[1]+=w,g[d+1]-=w;else f[j]+=d/a[j]*w,Sol(d%a[j],w);
}
int main(){
RI i,p;LL x;for(read(p,q),a[n=1]=p,i=1;i<=q;i++){read(x);W(n&&a[n]>=x) n--;a[++n]=x;}
for(f[n]=1,i=n-1;i>=1;i--) f[i]+=a[i+1]/a[i]*f[i+1],Sol(a[i+1]%a[i],f[i+1]);
for(g[1]+=f[1],g[a[1]+1]-=f[1],i=1;i<=p;i++) writeln(g[i]+=g[i-1]);return 0;
}