結果

問題 No.270 next_permutation (1)
ユーザー pekempeypekempey
提出日時 2015-08-22 00:04:03
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 2,087 bytes
コンパイル時間 1,312 ms
コンパイル使用メモリ 161,400 KB
実行使用メモリ 6,400 KB
最終ジャッジ日時 2024-07-18 12:22:45
合計ジャッジ時間 2,245 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 3 ms
5,376 KB
testcase_05 WA -
testcase_06 AC 4 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 3 ms
5,376 KB
testcase_09 AC 11 ms
5,376 KB
testcase_10 AC 68 ms
6,272 KB
testcase_11 AC 75 ms
6,400 KB
testcase_12 AC 76 ms
6,272 KB
testcase_13 AC 65 ms
6,144 KB
testcase_14 AC 67 ms
6,144 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#define rep(i, a) rep2 (i, 0, a)
#define rep2(i, a, b) for (int i = (a); i < (b); i++)
#define repr(i, a) repr2 (i, 0, a)
#define repr2(i, a, b) for (int i = (b) - 1; i >= (a); i--)
#define asn(a, b, c) fill_n(&(b), sizeof(a) / sizeof(b), c)
using namespace std;
typedef long long ll;
const ll inf = 1e9;
const ll mod = 1e9 + 7;

template <class BidirectionalIterator, class Compare>
int next_permutation2(BidirectionalIterator first, BidirectionalIterator last, Compare comp)
{
  if (first == last)
    return false;

  BidirectionalIterator i = first;
  ++i;

  if (i == last)
    return 0;

  i = last;
  --i;

  for(;;) {
    BidirectionalIterator ii = i;
    --i;
    if (comp(*i, *ii)) {
      BidirectionalIterator j = last;
      while (!comp(*i, *--j)) {}

      std::swap(*i, *j);
      std::reverse(ii, last);
      return i - first;
    }
    if (i == first) {
      std::reverse(first, last);
      return i - first;
    }
  }
}

template <class BidirectionalIterator>
int next_permutation2(BidirectionalIterator first, BidirectionalIterator last)
{
  typedef
    typename std::iterator_traits<BidirectionalIterator>::value_type
  value_type;

  return next_permutation2(first, last, std::less<value_type>());
}

ll N, K;
ll p[1000000], B[1000000];

struct BIT {
	int size;
	vector<ll> bit;
	BIT(int size) : size(size), bit(size + 1) {}
	void add(int k, ll x) {
		while (k <= size) {
			bit[k] += x;
			k += k & -k;
		}
	}
	ll sum(int k) {
		ll res = 0;
		while (k > 0) {
			res += bit[k];
			k -= k & -k;
		}
		return res;
	}
	void update(int k, ll x) {
		ll g = sum(k) - sum(k - 1);
		add(k, x - g);
	}
};

int main() {
	cin >> N >> K;
	rep (i, N) cin >> p[i];
	rep (i, N) cin >> B[i];

	BIT bit(N * 2);
	ll ans = 0;

	if (K == 0) {
		cout << 0 << endl;
		return 0;
	}

	rep (i, N) {
		bit.add(i + 1, abs(B[i] - p[i]));
	}
	ans += bit.sum(N + 10);

	rep (ii, K - 1) {
		int k = next_permutation2(p, p + N);	
		rep2 (i, max(0, k - 1), N) {
			bit.update(i + 1, abs(B[i] - p[i]));
		}
		ans += bit.sum(N + 10);
	}
	cout << ans << endl;

	return 0;
}
0