結果
問題 | No.742 にゃんにゃんにゃん 猫の挨拶 |
ユーザー |
|
提出日時 | 2018-10-05 22:04:32 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 493 ms / 2,500 ms |
コード長 | 2,055 bytes |
コンパイル時間 | 1,847 ms |
コンパイル使用メモリ | 177,972 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-10-12 13:56:48 |
合計ジャッジ時間 | 3,266 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 16 |
ソースコード
#define _USE_MATH_DEFINES#include "bits/stdc++.h"using namespace std;#define FOR(i,j,k) for(int (i)=(j);(i)<(int)(k);++(i))#define rep(i,j) FOR(i,0,j)#define each(x,y) for(auto &(x):(y))#define mp make_pair#define MT make_tuple#define all(x) (x).begin(),(x).end()#define debug(x) cout<<#x<<": "<<(x)<<endl#define smax(x,y) (x)=max((x),(y))#define smin(x,y) (x)=min((x),(y))#define MEM(x,y) memset((x),(y),sizeof (x))#define sz(x) (int)(x).size()#define RT returnusing ll = long long;using pii = pair<int, int>;using vi = vector<int>;using vll = vector<ll>;template<class Val>struct BinaryIndexedTree {int n;vector<Val> t;BinaryIndexedTree() {}BinaryIndexedTree(int _n) :n(_n + 1), t(_n + 1) {}void add(int k, Val val) {k++;while (k < n) {t[k] += val;k += (k&-k);}}void set(int k, Val val) {add(k, -sum(k, k + 1));add(k, val);}Val sum(int k) {Val r = 0;while (k > 0) {r += t[k];k -= (k&-k);}return r;}Val sum(int l, int r) {return sum(r) - sum(l);}};int N;int f(vector<pii> a) {BinaryIndexedTree<int> bit(N);sort(all(a));int re = 0;each(p, a) {re += bit.sum(p.second, N);bit.add(p.second, 1);}return re;}void solve() {cin >> N;vector<pii> le, ri, riri;rep(i, N) {int to;cin >> to;--to;if (i <= to) {le.emplace_back(i, to);} else {ri.emplace_back(i, to);riri.emplace_back(N - 1 - i, N - 1 - to);}}int ans = 0;each(l, le) {each(r, ri) {if (l.first <= r.first && l.second >= r.second) {ans++;}}}ans += f(le) + f(riri);cout << ans << endl;}int main() {ios::sync_with_stdio(false);cin.tie(0);cout << fixed << setprecision(15);solve();return 0;}