寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

Aizu Online Judge in C++ #Volume26 : 2601 - Evacuation Route

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/