結果

問題 No.1496 不思議な数え上げ
ユーザー kotatsugamekotatsugame
提出日時 2021-05-02 04:17:13
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 513 ms / 3,000 ms
コード長 1,068 bytes
コンパイル時間 770 ms
コンパイル使用メモリ 74,412 KB
実行使用メモリ 11,872 KB
最終ジャッジ日時 2023-09-27 18:13:47
合計ジャッジ時間 25,541 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,440 KB
testcase_01 AC 2 ms
5,420 KB
testcase_02 AC 2 ms
5,424 KB
testcase_03 AC 470 ms
11,532 KB
testcase_04 AC 470 ms
11,752 KB
testcase_05 AC 461 ms
11,744 KB
testcase_06 AC 452 ms
11,612 KB
testcase_07 AC 446 ms
11,692 KB
testcase_08 AC 449 ms
11,628 KB
testcase_09 AC 446 ms
11,872 KB
testcase_10 AC 448 ms
11,548 KB
testcase_11 AC 463 ms
11,744 KB
testcase_12 AC 478 ms
11,872 KB
testcase_13 AC 463 ms
11,608 KB
testcase_14 AC 456 ms
11,816 KB
testcase_15 AC 451 ms
11,600 KB
testcase_16 AC 454 ms
11,612 KB
testcase_17 AC 493 ms
11,608 KB
testcase_18 AC 495 ms
11,552 KB
testcase_19 AC 513 ms
11,532 KB
testcase_20 AC 497 ms
11,744 KB
testcase_21 AC 506 ms
11,812 KB
testcase_22 AC 510 ms
11,544 KB
testcase_23 AC 497 ms
11,608 KB
testcase_24 AC 508 ms
11,608 KB
testcase_25 AC 512 ms
11,536 KB
testcase_26 AC 506 ms
11,656 KB
testcase_27 AC 476 ms
11,680 KB
testcase_28 AC 477 ms
11,532 KB
testcase_29 AC 487 ms
11,740 KB
testcase_30 AC 474 ms
11,684 KB
testcase_31 AC 510 ms
11,608 KB
testcase_32 AC 438 ms
11,208 KB
testcase_33 AC 225 ms
8,968 KB
testcase_34 AC 211 ms
8,720 KB
testcase_35 AC 6 ms
5,576 KB
testcase_36 AC 447 ms
11,272 KB
testcase_37 AC 280 ms
9,148 KB
testcase_38 AC 225 ms
8,664 KB
testcase_39 AC 335 ms
10,564 KB
testcase_40 AC 179 ms
8,252 KB
testcase_41 AC 334 ms
10,648 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:12:1: 警告: ISO C++ では型の無い ‘main’ の宣言を禁止しています [-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