結果
| 問題 | 
                            No.929 よくあるボールを移動するやつ
                             | 
                    
| コンテスト | |
| ユーザー | 
                             yk356
                         | 
                    
| 提出日時 | 2019-11-22 23:07:09 | 
| 言語 | C++14  (gcc 13.3.0 + boost 1.87.0)  | 
                    
| 結果 | 
                             
                                RE
                                 
                             
                            
                         | 
                    
| 実行時間 | - | 
| コード長 | 839 bytes | 
| コンパイル時間 | 1,523 ms | 
| コンパイル使用メモリ | 173,088 KB | 
| 実行使用メモリ | 6,824 KB | 
| 最終ジャッジ日時 | 2024-10-11 04:43:30 | 
| 合計ジャッジ時間 | 5,742 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge5 / judge2 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 4 | 
| other | AC * 2 WA * 1 RE * 9 | 
ソースコード
#include <bits/stdc++.h>
using namespace std;
int main()
{
    int N;
    cin >> N;
    vector<pair<int,int>> cnt;
    vector<int> flag;
    for(int i = 0; i < N; i++)
    {
        int b;
        scanf("%d", &b);
        if(b == 0) flag.push_back(i);
        else if(b >= 2) cnt.push_back(make_pair(i, b-1));
    }
    long long ans = 0;
    for(int i = 0; i < flag.size(); i++)
    {
        int idx = flag[i];
        int jdx = 0;
        int minv = 1e8;
        for(int j = 0; j < cnt.size(); j++)
        {
            int c1 = cnt[j].first;
            int c2 = cnt[j].second;
            if(abs(idx-c1) < minv && c2 > 0)
            {
                jdx = c1;
                minv = abs(idx-c1);
            }
        }
        cnt[jdx].second -= 1;
        ans += (long long)minv;
    }
    cout << ans << endl;
    return 0;
}
            
            
            
        
            
yk356