2020牛客暑期多校训练营(第二场)G. Greater and Greater(bitset优化fft)

题目

一个长为n(n<=150000)的序列a,第i个数ai(1<=ai<=1e9)

一个长为m(m<=min(n,40000))的序列b,第j个数bj(1<=bj<=1e9)

求a中有多少长为m的子区间S,满足对应任意[1,m],Si>=bi

思路来源

夏老师的submission

题解

暴力是O(n*m)的,6e9,考虑引入bitset除掉一个64,复杂度就稳了

独立考虑a中的每个值,能大于哪些b中的值,也就是将a和b中的值放到一起排序

遇到b的值就给bitset上赋上一位,遇到a中的值就令a的答案等于当前的bitset的值

这样bitset本质上只会变化m次,赋n次值每次操作数是m/64,总复杂度O(n*m/64)

求出这n个数的bitset后,就是一个本质fft问题,

如果a的[i,i+m-1]需要和b的[0,m-1]比,就需要满足二者下标差为i

所以可以先将b序列每个值对应的bitset翻转,也就是令j=m-1-j,

这样i-j=k转化为i+m-1-j=k+m-1,

a中[i,i+m-1]和b中[0,m-1]做比较时,就需要到下标i+m-1上找

而最终下标上的值应该是一个与操作,即只要有一位为0不合法,该区间就不合法

由于bitset与/或时,两个bitset需要等长,无法控制一个长为n,另一个长为m

两个bitset都长为n时,就会有一些不合法的位,而左移新挪出来的位默认为0

所以,这里采用的是或的方式,即令有效位为0,

如果不存在非有效位,或在一起就应该是0,

统计最终bitset的[m-1,n-1]的0的个数即可

代码

#include
using namespace std;
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define per(i,a,b) for(int i=(a);i>=(b);--i)
typedef long long ll;
typedef double db;
typedef pair P;
#define fi first
#define se second
#define dbg(x) cerr<<(#x)<<":"< dist(l, r); return dist(gen); }
const int N=150010,M=40010,INF=0x3f3f3f3f;
int n,m,cnt;
struct node{
    int v,p,op;
}e[N+M];
bool operator<(node a,node b){
    if(a.v!=b.v)return a.vb.op;
}
bitsettmp,ans;
/*
i-j=k(0<=k<=n-m)
i+m-1-j=k(m-1<=k<=n-1)
*/
int main(){
    //freopen("edu.out","w",stdout);
    sci(n),sci(m);
    rep(i,0,n-1)sci(e[i].v),e[i].p=i,e[i].op=1;
    rep(i,n,n+m-1)sci(e[i].v),e[i].p=i-n,e[i].op=2;
    sort(e,e+n+m);
    rep(i,0,m-1)tmp.set(i);
    rep(i,0,n+m-1){
        int p=e[i].p;
        if(e[i].op==1){
            ans|=(tmp<

你可能感兴趣的:(#,FFT/NTT/FWT/FMT,模拟(暴力/大模拟),bitset,fft)