Aizu Online Judge(AOJ)の過去問から、その提出コードの解答例です。
・問題 "Evacuation Route"
https://onlinejudge.u-aizu.ac.jp/problems/2601
・避難経路
僕が作成、提出したコードは、以下のとおりです。
・Aizu Online Judge in C++ #Volume26 - 2601 : Evacuation Route
/* Aizu Online Judge in C++ #Volume26 - 2601 : Evacuation Route https://onlinejudge.u-aizu.ac.jp/problems/2601 提出コードの解答例 https://neguse-atama.hatenablog.com */ #include<bits/stdc++.h> using namespace std; using ll=long long; const ll N=1<<17; ll w,a[N],l[N],r[N]; int main(void){ cin>>w; for(ll i=0;i<w;i++){ cin>>a[i]; } l[0]=(a[0]==0 ? 1e9 : 0); for(ll i=1;i<w;i++){ l[i]=(a[i]==0 ? 1e9 : (a[i]<0 ? min(-a[i],max(l[i-1]-1,0LL)) : max(l[i-1]-1,0LL))); } r[w-1]=(a[w-1]==0 ? 1e9 : 0); for(ll i=w-1;i--;){ r[i]=(a[i]==0 ? 1e9 : (a[i]<0 ? min(-a[i],max(r[i+1]-1,0LL)) : max(r[i+1]-1,0LL))); } ll ans=0; for(ll i=0;i<w;i++){ if(a[i]>0){ ans+=min(a[i],max(l[i],r[i])); } } cout<<ans<<endl; return 0; }
設問の出典は、プログラミング問題のオンライン採点システム「Aizu Online Judge(AOJ)」です。
http://judge.u-aizu.ac.jp/onlinejudge/