結果

問題 No.929 よくあるボールを移動するやつ
ユーザー uryo1903
提出日時 2019-11-22 21:59:39
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 15 ms / 2,000 ms
コード長 1,244 bytes
コンパイル時間 1,668 ms
コンパイル使用メモリ 175,080 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-11 03:36:39
合計ジャッジ時間 2,447 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff #

//header{{{
#include<bits/stdc++.h>
  
using namespace std;
 
#define rep(i,n) for(int i=0;i<(n);++i)
#define reps(i,n) for(int i=1;i<=(n);++i)
#define all(x) (x).begin(),(x).end()
#define setout(n,x) setw(n) << setfill(x)
#define Fixed fixed << setprecision(10)
#define int long long
using pii = pair<int,int>;
constexpr int INF  = 0x3f3f3f3f;
constexpr long long LINF = 0x3f3f3f3f3f3f3f3fLL;
constexpr int mod = 1e9+7; 

template <class A, class B> inline bool chmax(A &a, const B &b) { return b > a && (a = b, true); }
template <class A, class B> inline bool chmin(A &a, const B &b) { return b < a && (a = b, true); }

template <class T> using min_heap = priority_queue<T,vector<T>,greater<T> >;
template <class T> using max_heap = priority_queue<T>;
 
int gcd(int a,int b){ return b ? gcd(b,a % b) : a;}
int lcm(int a,int b){ return a / gcd(a,b) * b;}
//}}}

signed main(){
	cin.tie(0);
	ios::sync_with_stdio(false);
	int n;
	cin >> n;
	vector<int> b(n);
	min_heap<pii> que;
	rep(i,n){
		cin >> b[i];
		if(b[i] > 1) que.emplace(i,b[i]);
	}
	int cnt = 0;
	rep(i,n){
		if(b[i] == 0){
			pii x = que.top();
			que.pop();
			cnt += abs(i - x.first);
			x.second--;
			if(x.second > 1) que.emplace(x);
		}
	}
	cout << cnt << '\n';
	return 0;
}

0