赛时想出来但是没写出来。
题意简述
定义一个合法序列 \(a_1,a_2,\cdots a_m\) 满足以下条件:
- 对于 \(1\le i<m\),\(a_{i+1}\) 被 \(a_i\) 整除。
初始给定排列 \(a\),\(q\) 次询问,每次询问给出 \(l,r\),求出 \(a_{[l,r]}\) 的所有可以不连续的子序列中,合法的序列的数目。
\(n,q\le 10^6\),时间限制 6s。
分析
如果我们知道的原本的序列,考虑 dp。设 \(f_i\) 表示以 \(i\) 结尾的合法序列个数,转移方程即为 \(f_i=1+\sum_{1\le j<i,a_j|a_i}f_j\)。因为原序列是一个排列,保证了没有相同的数字,所以考虑直接枚举 \(a_j\) 的倍数进行转移,时间复杂度是调和级数的。
现在有了区间查询。我们先用个数据结构维护 dp 值,下文出于某些原因,\(f_i\) 是以 \(i\) 为开头的合法序列个数。
考虑离线,从后往前枚举左端点 \(l\),\(f_l\) 的值可以直接枚举倍数求,时间复杂度调和级数。
考虑如何去除掉包含 \((r,n]\) 范围内的合法序列。我们发现沿用 \(f\) 的状态定义难以去除掉所有不在范围内的合法序列,所以我们新设 \(g_i\) 表示以 \(i\) 为结尾的合法序列(同时也用数据结构维护),容易发现这样的话一次查询的答案就是 \(\sum_{i=l}^nf_i-\sum_{i=r+1}^ng_i\)(因为 \(g_i\) 同时也计算了所有数都在 \((r,n]\) 范围内的序列,所以我们要把 \((r,n]\) 范围内的 \(f_i\) 也加进来)。
现在考虑如何在左移左端点 \(l\) 时计算 \(g_i\)。显然只有 \(a_l\) 的倍数才会被更新 \(g_i\)。所以同样枚举 \(a_l\) 的倍数,首先对这些数记录一个 \(delta_i\) 表示这次更新中 \(g_i\) 的变化量,初始化为 \(1\)(因为无论如何该数和 \(a_l\) 构成一个合法序列,当然初始化为 \(0\) 也可以做)。然后对每个数开一个 vector 记录它的约数。枚举 \(a_i\) 的每个约数 \(a_j\),若 \(a_l|a_j\),则将 \(delta_i\leftarrow delta_i+delta_j\),最后将 \(delta_i\) 加进维护 \(g\) 的数据结构里。
那么这道题就做完了,我们发现该数据结构需要支持单点加和后缀求和,树状数组即可。
时间复杂度大体是 \(O(n\log n\cdot d(n))\) 的,但这个 log 和这个 \(d(n)\) 都是远远跑不满的。关于卡常,优化一下读入,再加个 inline
,就能过了。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<map>
#include<unordered_map>
#include<vector>
#include<queue>
#include<bitset>
#include<set>
#include<ctime>
#include<random>
#define x1 xx1
#define y1 yy1
#define IOS ios::sync_with_stdio(false)
#define ITIE cin.tie(0);
#define OTIE cout.tie(0);
#define PY puts("Yes")
#define PN puts("No")
#define PW puts("-1")
#define P__ puts("")
#define PU puts("--------------------")
#define popc __builtin_popcount
#define mp make_pair
#define fi first
#define se second
#define gc getchar
#define pc putchar
#define pb emplace_back
#define rep(a,b,c) for(int a=(b);a<=(c);++a)
#define per(a,b,c) for(int a=(b);a>=(c);--a)
#define reprange(a,b,c,d) for(int a=(b);a<=(c);a+=(d))
#define perrange(a,b,c,d) for(int a=(b);a>=(c);a-=(d))
#define graph(i,j,k,l) for(int i=k[j];i;i=l[i].nxt)
#define lowbit(x) (x&-x)
#define lson(x) (x<<1)
#define rson(x) (x<<1|1)
#define mem(x,y) memset(x,y,sizeof x)
//#define double long double
//#define int long long
//#define int __int128
using namespace std;
typedef long long i64;
using pii=pair<int,int>;
bool greating(int x,int y){return x>y;}
bool greatingll(long long x,long long y){return x>y;}
inline int rd(){
int 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-48;ch=getchar();}return x*f;
}
inline void write(i64 x,char ch='\0'){
if(x<0){x=-x;putchar('-');}
int y=0;char z[40];
while(x||!y){z[y++]=x%10+48;x/=10;}
while(y--)putchar(z[y]);if(ch!='\0')putchar(ch);
}
bool Mbg;
const int maxn=1e6+5,maxm=4e5+5,inf=0x3f3f3f3f;
const long long llinf=0x3f3f3f3f3f3f3f3f;
int n,Q,a[maxn],pos[maxn];
i64 ans[maxn];
vector<pii>q[maxn];
vector<int>v[maxn];
i64 f[maxn],g[maxn];
struct BIT{
i64 c[maxn];
void init(){rep(i,1,n)c[i]=0;}
inline void add(int x,i64 y){x=n-x+1;while(x<=n)c[x]+=y,x+=lowbit(x);}
inline i64 qry(int x){i64 res=0;x=n-x+1;while(x)res+=c[x],x-=lowbit(x);return res;}
}A,B;
void init(){
A.init(),B.init();
rep(i,1,n)q[i].clear(),v[i].clear();
}
void solve_the_problem(){
cin>>n>>Q,init();
rep(i,1,n)cin>>a[i],pos[a[i]]=i;
rep(i,1,Q){
int l,r;cin>>l>>r;
q[l].pb(mp(r,i));
}
per(l,n,1){
int x=a[l];
f[l]=1;
reprange(y,x*2,n,x)if(pos[y]>l){
f[l]+=f[pos[y]],g[pos[y]]=1;
for(int u:v[pos[y]])if(a[u]%a[l]==0)g[pos[y]]+=g[u];
B.add(pos[y],g[pos[y]]),v[pos[y]].pb(l);
}
A.add(l,f[l]),B.add(l,1);
for(pii i:q[l]){
int r=i.fi,id=i.se;
ans[id]=A.qry(l)-B.qry(r+1);
}
}
rep(i,1,Q)cout<<ans[i]<<' ';cout<<'\n';
}
bool Med;
signed main(){
// freopen(".in","r",stdin);freopen(".out","w",stdout);
// fprintf(stderr,"%.3lfMB\n",(&Mbg-&Med)/1048576.0);
IOS;ITIE;OTIE;int _;cin>>_;while(_--)solve_the_problem();
}
/*
*/
标签:int,Nobody,CF1946F,pos,needed,delta,序列,include,define
From: https://www.cnblogs.com/dcytrl/p/18097808