結果

問題 No.270 next_permutation (1)
コンテスト
ユーザー pekempey
提出日時 2017-01-19 16:05:00
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 29 ms / 2,000 ms
コード長 1,510 bytes
コンパイル時間 833 ms
コンパイル使用メモリ 72,960 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-23 01:30:34
合計ジャッジ時間 2,139 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>

using namespace std;

int main() {
	int n, K;
	cin >> n >> K;

	vector<int> a(n);
	for (int i = 0; i < n; i++) {
		scanf("%d", &a[i]);
	}

	long long d = 0;

	vector<int> b(n);
	for (int i = 0; i < n; i++) {
		scanf("%d", &b[i]);
		d += abs(b[i] - a[i]);
	}

	long long ans = 0;

	while (K--) {
		ans += d;

		// next_permutation
		// 正当性は明らかなので、計算量解析だけおこなう。
		// 
		// 解析にはポテンシャルを用いる。
		// ポテンシャル関数は "Φ:=a[i]>a[i+1] となる i の総数"と定義する。
		// 
		// #1の while が回った回数を X とする。
		// このとき、⊿Φ=1-Xである。なぜなら、
		// #2のswapによりΦは+1、
		// #3のreverseによりΦは-Xするからである。
		//
		// ならしコストは、(whileが回った回数)+(ポテンシャルの変化量)とみなせるので、
		// ならしコストは X+(1-X)=1 となる。

		// #1
		int i = n - 2;
		while (i >= 0 && a[i] >= a[i + 1]) i--;

		if (i == -1) {
			reverse(a.begin(), a.end());
			d = 0;
			for (int i = 0; i < n; i++) {
				d += abs(b[i] - a[i]);
			}
			continue;
		}

		int j = i + 1;
		while (j + 1 < n && a[j + 1] > a[i]) j++;

		for (int k = i; k < n; k++) {
			d -= abs(b[k] - a[k]);
		}

		// #2
		swap(a[i], a[j]);

		// #3
		reverse(a.begin() + i + 1, a.end());

		for (int k = i; k < n; k++) {
			d += abs(b[k] - a[k]);
		}
	}

	cout << ans << endl;
}
0