NKOI Round 8（12.14）

NKOI Round 8
T1.呼叫机房的同学（组合数学）

T2.突然点亮的故障灯（tarjan缩点+拓扑序dp+线段树优化建图）

T3.灵异电梯（状压dp）

T1

这题神啊。首先我们可以O(n)预处理组合数（O(n)算出所有逆元，O(n)做两个前缀积）。然后我们发现可以$O(n^2)$ 算一个点到一个矩形内的所有点的方案数之和。就是一堆组合数加起来。然后我们发现可以一行一行的算，每一行都是$C_a^b+C_{a+1}^{b}+...+C_{a+c}^b$，通过神奇的变化，他就等于$C_{a+c+1}^{b+1}-C_{a}^{b+1}$，然后我们就可以O(n)计算了。然后再划一下，发现其实可以O(1)计算。是四项组合数的加加减减。然后这题复杂度就是$O(n^2)$啦。那怎么变成O(n)呢？我们枚举b中的每个点，四项组合数乘上四项组合数，应该有十六类。我们分别计算每一类，然后加加减减就是答案了。我们发现每一类组合数的意义其实可以变成两个定点到b矩形的路径的方案的和。然后再差分一下，变成一个点到一个矩形的左下角的方案数，然后可以枚举是从哪条边出的矩形来计算，变成O(n)，所以总的复杂度就是O(16*4*n).

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
#define mod 998244353
inline int read(){
    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-'0',ch=getchar();
    return x*f;
}
int n,rx[8],ry[8],inv[N<<1],fac[N<<1],facni[N<<1],ans=0;
int px[10],py[10];
inline int C(int x,int y){
    return (ll)fac[x]*facni[y]%mod*facni[x-y]%mod;
}
inline int calcc(int x1,int y1,int x2,int y2,int x3,int y3){
    int res=0;
    for(int i=x1;i<=x2;++i) 
        res=(res+(ll)C(i-x1+y2-y1,i-x1)*C(x3-i+y3-y2-1,x3-i)%mod*(i-x1+y2-y1+1))%mod;
    for(int i=y1;i<=y2;++i) 
        res=(res+(ll)C(x2-x1+i-y1,i-y1)*C(x3-x2-1+y3-i,y3-i)%mod*(x2-x1+i-y1+1))%mod;
    return res;
}
inline int calc(int x2,int y2,int x1,int y1){
    int res=calcc(x1,y1,rx[4],ry[4],x2,y2)-calcc(x1,y1,rx[3]-1,ry[4],x2,y2);res%=mod;
    res=res-calcc(x1,y1,rx[4],ry[3]-1,x2,y2)+calcc(x1,y1,rx[3]-1,ry[3]-1,x2,y2);res%=mod;
    if(res<0) res+=mod;return res;
}
int main(){
    freopen("student.in","r",stdin);
    freopen("student.out","w",stdout);
    n=read();for(int i=1;i<=6;++i) rx[i]=read();
    for(int i=1;i<=6;++i) ry[i]=read();inv[1]=1;fac[0]=facni[0]=1;
    for(int i=2;i<=n<<1;++i) inv[i]=(ll)inv[mod%i]*(mod-mod/i)%mod;
    for(int i=1;i<=n<<1;++i) fac[i]=(ll)fac[i-1]*i%mod,facni[i]=(ll)facni[i-1]*inv[i]%mod;
    px[1]=rx[6]+1;py[1]=ry[6]+1;px[2]=rx[6]+1;py[2]=ry[5];
    px[3]=rx[5];py[3]=ry[5];px[4]=rx[5];py[4]=ry[6]+1;
    px[5]=rx[1]-1;py[5]=ry[1]-1;px[6]=rx[1]-1;py[6]=ry[2];
    px[7]=rx[2];py[7]=ry[2];px[8]=rx[2];py[8]=ry[1]-1;
    for(int i=1;i<=4;++i)
        for(int j=5;j<=8;++j){
            ans+=calc(px[i],py[i],px[j],py[j])*(i&1 ^ j&1?-1:1);
            ans%=mod;
        }if(ans<0) ans+=mod;printf("%d\n",ans);
    return 0;
}

T2

题解见：传送门

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <stack>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 500010
inline int read(){
    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-'0',ch=getchar();
    return x*f;
}
int n,pos[N],L1[N<<2],R1[N<<2],h[N<<2],num=0,tot=0,in[N<<2],q[N<<2],h1=0,t=0;
int dfn[N<<2],low[N<<2],dfnum=0,scc,bel[N<<2],L[N<<2],R[N<<2],id[N];
bool inq[N<<2];
struct edge{
    int fr,to,next;
}data[N*20];
inline void add(int x,int y){
    if(x==y) return;
    data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].fr=x;
}
inline void build(int p,int l,int r){
    tot=max(tot,p);L1[p]=l;R1[p]=r;
    if(l==r){id[l]=p;return;}int mid=l+r>>1;
    build(p<<1,l,mid);build(p<<1|1,mid+1,r);
    add(p,p<<1);add(p,p<<1|1);
}
inline void ask(int p,int l,int r,int x,int y,int s){
    if(x<=l&&r<=y){add(s,p);return;}
    int mid=l+r>>1;
    if(x<=mid) ask(p<<1,l,mid,x,y,s);
    if(y>mid) ask(p<<1|1,mid+1,r,x,y,s);
}
stack<int>qq;
inline void tarjan(int x){
    dfn[x]=low[x]=++dfnum;qq.push(x);inq[x]=1;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;
        if(!dfn[y]) tarjan(y),low[x]=min(low[x],low[y]);
        else if(inq[y]) low[x]=min(low[x],dfn[y]);
    }if(dfn[x]==low[x]){
        ++scc;L[scc]=inf;
        while(1){
            int y=qq.top();qq.pop();inq[y]=0;bel[y]=scc;
            L[scc]=min(L[scc],L1[y]);R[scc]=max(R[scc],R1[y]);
            if(y==x) break;
        }
    }
}
int main(){
    freopen("light.in","r",stdin);
    freopen("light.out","w",stdout);
    n=read();for(int i=1;i<=n;++i) pos[i]=read();build(1,1,n);
    for(int i=1;i<=n;++i){
        int x=read();int l=lower_bound(pos+1,pos+n+1,pos[i]-x)-pos;
        int r=upper_bound(pos+1,pos+n+1,pos[i]+x)-pos-1;
        if(l==r) continue;ask(1,1,n,l,r,id[i]);
    }for(int i=1;i<=tot;++i) if(!dfn[i]) tarjan(i);memset(h,0,sizeof(h));
    for(int i=1;i<=num;++i){
        int x=data[i].fr,y=data[i].to;
        if(bel[x]==bel[y]) continue;in[bel[y]]++;
        data[i].fr=bel[x];data[i].to=bel[y];data[i].next=h[bel[x]];h[bel[x]]=i;
    }for(int i=1;i<=scc;++i) if(!in[i]) q[++t]=i;
    while(h1<t){
        int x=q[++h1];
        for(int i=h[x];i;i=data[i].next){
            int y=data[i].to;if(--in[y]==0) q[++t]=y;
        }
    }for(int ii=scc;ii>=1;--ii){
        int x=q[ii];
        for(int i=h[x];i;i=data[i].next){
            int y=data[i].to;L[x]=min(L[x],L[y]);R[x]=max(R[x],R[y]);
        }
    }for(int i=1;i<=n;++i) printf("%d\n",R[bel[id[i]]]-L[bel[id[i]]]+1);
    return 0;
}