結果
問題 | No.1496 不思議な数え上げ |
ユーザー | penguinman |
提出日時 | 2021-03-30 06:49:44 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 738 ms / 3,000 ms |
コード長 | 1,484 bytes |
コンパイル時間 | 2,185 ms |
コンパイル使用メモリ | 202,028 KB |
最終ジャッジ日時 | 2025-01-20 00:42:27 |
ジャッジサーバーID (参考情報) |
judge5 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 39 |
ソースコード
#include<bits/stdc++.h>using std::cin;using std::cout;using std::endl;using std::vector;using ll = long long;#define rep(i,j,k) for(int i=int(j); i<int(k); i++)#define REP(i,j,k) for(int i=int(j); i<=int(k); i++)int main(){ll N; cin >> N;vector<ll> P(N);rep(i,0,N) cin >> P[i];vector<ll> sum(N+1), rev(N);rep(i,0,N){sum[i+1] = sum[i]+P[i];rev[--P[i]] = i;}std::set<ll> set;set.insert(-1);set.insert(N);rep(i,0,N){ll A; cin >> A;auto itr = set.lower_bound(rev[i]);ll right = *itr;itr--;ll left = *itr;ll ans = 0;if(right-rev[i] < rev[i]-left){rep(r,rev[i],right){//右端固定ll l = std::lower_bound(sum.begin(),sum.end(),sum[r+1]-A)-sum.begin();//left < l ≤ rev[i] を満たす必要があるl = std::max(l, left+1);l = std::min(l, rev[i]+1);ans += rev[i]-l+1;}}else{REP(l,left+1,rev[i]){//左端固定ll r = std::upper_bound(sum.begin(),sum.end(),sum[l]+A)-sum.begin()-2;//rev[i] ≤ r < right を満たす必要があるr = std::min(r, right-1);r = std::max(r, rev[i]-1);ans += r-rev[i]+1;}}cout << ans << "\n";set.insert(rev[i]);}}