結果

問題 No.270 next_permutation (1)
ユーザー ctyl_0ctyl_0
提出日時 2015-10-01 00:27:54
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 2,268 bytes
コンパイル時間 752 ms
コンパイル使用メモリ 89,428 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-19 18:46:35
合計ジャッジ時間 1,775 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 3 ms
5,376 KB
testcase_02 AC 3 ms
5,376 KB
testcase_03 WA -
testcase_04 AC 3 ms
5,376 KB
testcase_05 AC 3 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 9 ms
5,376 KB
testcase_10 AC 61 ms
5,376 KB
testcase_11 AC 62 ms
5,376 KB
testcase_12 AC 62 ms
5,376 KB
testcase_13 AC 58 ms
5,376 KB
testcase_14 AC 57 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <iomanip>
#include <vector>
#include <algorithm>
#include <numeric>
#include <functional>
#include <cmath>
#include <queue>
#include <stack>
#include <set>
#include <map>

#define repd(i,a,b) for (int i=(a);i<(b);i++)
#define rep(i,n) repd(i,0,n)
#define pb(x) push_back(x)
#define mp(x,y) make_pair((x),(y))

typedef long long ll;

using namespace std;


int inputValue(){
    int a;
    cin >> a;
    return a;
};

template <typename T>
void output(T a, int precision) {
    if(precision > 0){
        cout << setprecision(precision)  << a << "\n";
    }
    else{
        cout << a << "\n";
    }
}

void my_next_permutation(vector<int> &p, vector<pair<int, int>> &swp){
    swp.clear();
    if (p.size() == 1) {
        return;
    }
    int s = (int)p.size();
    int k = s - 2;
    int l = s - 1;
    for (; k >= 0; k--) {
        if (p[k + 1] > p[k]) {
            break;
        }
        if (k == 0) {
            for (; k < l; k++, s--) {
                //swap(p[k], p[l]);
                swp.push_back(mp(k, l));
            }
            return;
        }
    }
    
    
    for (; l > k; l--) {
        if (p[l] > p[k]) {
            break;
        }
    }
    
    //swap(p[k], p[l]);
    swp.push_back(mp(k, l));
    
    for (; k + 1 < s - 1; k++, s--) {
        //swap(p[k + 1], p[s - 1]);
        swp.push_back(mp(k + 1, s - 1));
    }
    
}

int main() {
    
    // source code
    int N = inputValue();
    int K = inputValue();
    vector<int> P(N);
    vector<int> B(N);
    
    rep(i, N){
        P[i] = inputValue();
    }
    
    rep(i, N){
        B[i] = inputValue();
    }
    
    ll ret = 0;
    rep(i, N){
        ret += abs(P[i] - B[i]);
    }
    
    ll dist = ret;
    
    vector<pair<int, int>> swp;
    
    rep(k, K - 1){
        my_next_permutation(P, swp);
        
        rep(i, swp.size()){
            dist -= abs(P[swp[i].first] - B[swp[i].first]);
            dist -= abs(P[swp[i].second] - B[swp[i].second]);
            swap(P[swp[i].first], P[swp[i].second]);
            dist += abs(P[swp[i].first] - B[swp[i].first]);
            dist += abs(P[swp[i].second] - B[swp[i].second]);
        }
        
        ret+= dist;

    }
    
    output(ret, 0);
        
    
    return 0;
}
0