結果

問題 No.370 道路の掃除
ユーザー hanorver
提出日時 2016-05-14 00:35:39
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 7 ms / 2,000 ms
コード長 2,098 bytes
コンパイル時間 673 ms
コンパイル使用メモリ 66,940 KB
実行使用メモリ 7,424 KB
最終ジャッジ日時 2024-10-14 16:52:47
合計ジャッジ時間 1,761 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<cstdlib>
#include<algorithm>
#include<climits>

int main(){
    int n, m;
    std::cin >> n >> m;

    std::vector<int> dist(m);
    for(int i = 0; i < m; i++){
        std::cin >> dist[i];
    }
    dist.push_back(0);

    std::sort(dist.begin(), dist.end());

    int ddist[1001][1001] = {0};
    for(size_t i = 0; i < dist.size(); i++){
        for(size_t j = 0; j < dist.size(); j++){
            ddist[i][j] = std::abs(dist[i] - dist[j]);
        }
    }

    int start = 0;
    for(size_t i = 0; i < dist.size(); i++){
        if(dist[i] == 0){
            start = i;
            break;
        }
    }

    long long min_cost = LLONG_MAX;

    std::vector<int> right;
    right.push_back(0);
    long long right_sum = 0;
    for(int i = start; i < dist.size() && i - start <= n; i++){
        int count = i - start;
        if(i - start == n){
            min_cost = std::min(min_cost, right_sum);
            break;
        }
        long long left_sum = 0;
        if(start - 1 >= 0) {
            left_sum = ddist[i][start - 1];
            count++;
        }
        for(int j = start - 2; j >= 0 && count < n; j--){
            left_sum += ddist[j + 1][j];
            count++;
        }
        if(count ==  n){
            min_cost = std::min(min_cost, right_sum + left_sum);
        }
        right_sum += ddist[i][i + 1];
    }

    right_sum = 0;
    for(int i = start; i >= 0 && i - start <= n; i--){
        int count = start - i;
        if(start - i == n){
            min_cost = std::min(min_cost, right_sum);
            break;
        }
        long long left_sum = 0;
        if(start - 1 >= 0) {
            left_sum = ddist[i][start + 1];
            count++;
        }
        for(int j = start + 2; j < dist.size() && count < n; j++){
            left_sum += ddist[j - 1][j];
            count++;
        }
        if(count ==  n){
            min_cost = std::min(min_cost, right_sum + left_sum);
        }
        if(i - 1 >= 0) right_sum += ddist[i][i - 1];
    }


    std::cout << min_cost << std::endl;
    return 0;
}
0