結果
問題 | No.1496 不思議な数え上げ |
ユーザー | りあん |
提出日時 | 2021-04-30 22:30:57 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 264 ms / 3,000 ms |
コード長 | 1,426 bytes |
コンパイル時間 | 2,177 ms |
コンパイル使用メモリ | 201,460 KB |
最終ジャッジ日時 | 2025-01-21 03:35:09 |
ジャッジサーバーID (参考情報) |
judge2 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 39 |
ソースコード
#include<bits/stdc++.h>using namespace std;using P = pair<int, int>;const int M = 1000000007;// const int M = 998244353;const long long LM = 1LL << 60;int main() {cin.tie(0);ios::sync_with_stdio(0);int n;cin >> n;vector<int> p(n), idx(n);vector<long long> a(n);for (int i = 0; i < n; ++i) {cin >> p[i];idx[p[i] - 1] = i;}for (int i = 0; i < n; ++i) {cin >> a[i];}vector<long long> imos(n + 1);for (int i = 0; i < n; ++i) {imos[i + 1] = imos[i] + p[i];}set<int> se;se.insert(-1);se.insert(n);for (int k = 0; k < n; ++k) {long long ans = 0;int i = idx[k];int l, r;{auto ptr = se.insert(i).first;l = *prev(ptr) + 1;r = *next(ptr);}int dl = i - l + 1;int dr = r - i;if (dl <= dr) {for (int j = l; j <= i; ++j) {auto ptr = upper_bound(imos.begin(), imos.begin() + r + 1, a[k] + imos[j]);ans += max(ptr - (imos.begin() + i) - 1, 0L);}}else {for (int j = i + 1; j <= r; ++j) {auto ptr = lower_bound(imos.begin() + l, imos.end(), imos[j] - a[k]);ans += max((imos.begin() + i) - ptr + 1, 0L);}}cout << ans << '\n';}return 0;}