51nod 1962 區間計數(單調棧+二分)
阿新 • • 發佈:2017-09-23
urn amp 單調棧 one pac span img top spa
維護兩個單調遞減的棧,當i加進棧,位置x的數彈出的時候,在另一個棧中找到和這個數一樣大的數,計算貢獻(x-靠右左端點)*(i-x)。
#include<iostream> #include<cstring> #include<cstdlib> #include<cstdio> #include<algorithm> #define ll long long using namespace std; const int maxn=500010,inf=1e9; int n,m,x,y,z,tot,topa,topb;View Codeint a[maxn],b[maxn],sta[maxn],stb[maxn]; ll ans; void read(int &k) { int f=1;k=0;char c=getchar(); while(c<‘0‘||c>‘9‘)c==‘-‘&&(f=-1),c=getchar(); while(c<=‘9‘&&c>=‘0‘)k=k*10+c-‘0‘,c=getchar(); k*=f; } int main() { read(n); for(int i=1;i<=n;i++)read(a[i]);for(int i=1;i<=n;i++)read(b[i]); a[++n]=inf;b[n]=inf-1; for(int i=1;i<=n;i++) { for(;topa&&a[sta[topa]]<=a[i];topa--) { if(!topb)continue; int l=1,r=topb; while(l<r) { int mid=(l+r)>>1; if(b[stb[mid]]<=a[sta[topa]])r=mid; else l=mid+1; } int x=l;if(b[stb[x]]!=a[sta[topa]])continue; ans+=1ll*max(0,min(stb[x],sta[topa])-max(stb[x-1],sta[topa-1]))*(i-max(stb[x],sta[topa])); } for(;topb&&b[stb[topb]]<=b[i];topb--) { if(!topa)continue; int l=1,r=topa; while(l<r) { int mid=(l+r)>>1; if(a[sta[mid]]<=b[stb[topb]])r=mid; else l=mid+1; } int x=l;if(a[sta[x]]!=b[stb[topb]])continue; ans+=1ll*max(0,min(sta[x],stb[topb])-max(sta[x-1],stb[topb-1]))*(i-max(sta[x],stb[topb])); } sta[++topa]=stb[++topb]=i; } printf("%lld\n",ans); return 0; }
51nod 1962 區間計數(單調棧+二分)