洛谷.5300.[GXOI/GZOI2019]與或和(單調棧)

LOJ
BZOJ
洛谷
php

想了一個奇葩的單調棧,算的時候要在中間取\(\min\),感受不靠譜不寫了=-=
調了十分鐘發現輸出沒取模=v=git

BZOJ好逗逼啊 題面連pdf都不掛了 哈哈哈哈spa


枚舉每一位。在這一位上與以後獲得\(1\)的就是全\(1\)子矩形個數。或以後獲得\(1\)的就是總舉行個數減去全\(0\)子矩形個數。
單調棧算一下就好啦。
維護一個單調遞增的棧。若是在右下角統計貢獻,每次遇到往上延伸長度\(l\leq sk[top]\)時,會把\(sk[top]\)的一部分截斷。用個變量\(sum\)維護前面全部柱子當前的延伸距離便可。
算惹我不說惹 看代碼叭= =code


//9208kb    7720ms
#include <cstdio>
#include <cctype>
#include <algorithm>
#define mod 1000000007
//#define gc() getchar()
#define MAXIN 500000
#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=1005;

int A[N][N];
char IN[MAXIN],*SS=IN,*TT=IN;

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-48,c=gc());
    return now;
}
int Calc(const int n,const int bit,const int equ)
{
    static int sk[N],sz[N],mx[N][N];
    LL ans=0; sk[0]=-1;
    for(int i=1; i<=n; ++i)
    {
        for(int j=1; j<=n; ++j) mx[i][j]=((A[i][j]>>bit&1)==equ)?mx[i-1][j]+1:0;
        LL sum=0;
        for(int j=1,top=0; j<=n; ++j)
        {
            int tot=0;
            while(sk[top]>=mx[i][j]) sum-=sk[top]*sz[top], tot+=sz[top--];
            sk[++top]=mx[i][j], sz[top]=tot+1;
            sum+=sk[top]*sz[top], ans+=sum;
        }
    }
    return ans%mod;
}

int main()
{
    const int n=read(); int mx=0;
    for(int i=1; i<=n; ++i)
        for(int j=1; j<=n; ++j) mx=std::max(mx,A[i][j]=read());
    LL ans1=0,ans2=0,tmp=(1ll*n*(n+1)*n*(n+1)>>2)%mod;
    for(int i=0; 1ll<<i<=mx; ++i) ans1+=(1ll<<i)*Calc(n,i,1)%mod;//1ll<<i<=mx!
    for(int i=0; 1ll<<i<=mx; ++i) ans2+=(1ll<<i)*(tmp+mod-Calc(n,i,0))%mod;
    printf("%lld %lld\n",ans1%mod,ans2%mod);

    return 0;
}
相關文章
相關標籤/搜索