結果

問題 No.1496 不思議な数え上げ
コンテスト
ユーザー kotatsugame
提出日時 2021-05-02 04:17:13
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 493 ms / 3,000 ms
コード長 1,068 bytes
コンパイル時間 677 ms
コンパイル使用メモリ 78,780 KB
実行使用メモリ 11,032 KB
最終ジャッジ日時 2024-07-20 12:43:00
合計ジャッジ時間 22,363 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 39
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:12:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   12 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
#include<atcoder/segtree>
using namespace std;
int N,P[2<<17];
int mx(int a,int b){return a<b?b:a;}
int mn(int a,int b){return a<b?a:b;}
int emx(){return -1;}
int emn(){return N;}
long S[2<<17];
int L[2<<17],R[2<<17];
int inv[2<<17];
main()
{
	cin>>N;
	for(int i=0;i<N;i++)
	{
		cin>>P[i];
		inv[P[i]-1]=i;
		S[i+1]=S[i]+P[i];
	}
	{
		atcoder::segtree<int,mx,emx>Q(N);
		for(int i=0;i<N;i++)
		{
			L[i]=Q.prod(0,P[i]);
			Q.set(P[i]-1,i);
		}
	}
	{
		atcoder::segtree<int,mn,emn>Q(N);
		for(int i=N;i--;)
		{
			R[i]=Q.prod(0,P[i]);
			Q.set(P[i]-1,i);
		}
	}
	for(int i=0;i<N;i++)
	{
		long A;cin>>A;
		int k=inv[i];
		long ans=0;
		if(k-L[k]<R[k]-k)
		{
			for(int x=L[k]+1;x<=k;x++)
			{
				int l=k,r=R[k]+1;
				while(r-l>1)
				{
					int mid=(l+r)/2;
					if(S[mid]-S[x]<=A)l=mid;
					else r=mid;
				}
				ans+=l-k;
			}
		}
		else
		{
			for(int y=k;y<R[k];y++)
			{
				int l=L[k]-1,r=k;
				while(r-l>1)
				{
					int mid=(l+r)/2;
					if(S[y+1]-S[mid+1]<=A)r=mid;
					else l=mid;
				}
				ans+=k-r;
			}
		}
		cout<<ans<<endl;
	}
}
0