結果

問題 No.2665 Minimize Inversions of Deque
コンテスト
ユーザー ゆにぽけ
提出日時 2024-03-08 21:35:00
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 59 ms / 2,000 ms
コード長 1,997 bytes
コンパイル時間 1,196 ms
コンパイル使用メモリ 129,064 KB
最終ジャッジ日時 2025-02-20 02:13:50
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <array>
#include <iterator>
#include <string>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <cassert>
#include <cmath>
#include <ctime>
#include <iomanip>
#include <numeric>
#include <stack>
#include <queue>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <bitset>
#include <random>
#include <utility>
#include <functional>
using namespace std;
template<class T> struct BinaryIndexedTree
{
	private:
	int N;
	vector<T> node;

	public:

	BinaryIndexedTree() : N(0) {}
	BinaryIndexedTree(int N) : N(N), node(N) {}

	void add(int pos,T x)
	{
		assert(0 <= pos && pos < N);
		pos++;
		while(pos <= N)
		{
			node[pos-1] += x;
			pos += pos & -pos;
		}
	}

	T sum(int l,int r)
	{
		assert(0 <= l && l <= N && r <= N);
		return sum(r) - sum(l);
	}

	T sum(int r)
	{
		T ret = 0;
		while(r > 0)
		{
			ret += node[r-1];
			r -= r & -r;
		}
		return ret;
	}

	T sum() {return sum(N);}
	
};

void Main()
{
	int N;
	cin >> N;
	vector<int> P(N);
	for(int i = 0;i < N;i++)
	{
		cin >> P[i];
		P[i]--;
	}
	vector<int> F(N);
	{
		BinaryIndexedTree<int> BIT(N);
		BIT.add(P[0],1);
		int prev = P[0];
		for(int i = 1;i < N;i++)
		{
			int u = BIT.sum(P[i],N);
			int l = BIT.sum(0,P[i]);
			if(l < u)
			{
				F[i] = 1;
				prev = P[i];
			}
			else if(l == u)
			{
				if(P[i] < prev)
				{
					F[i] = 1;
					prev = P[i];
				}
			}
			BIT.add(P[i],1);
		}
	}
	vector<int> ans;
	for(int i = N - 1;i >= 0;i--)
	{
		if(F[i])
		{
			ans.push_back(P[i]);
		}
	}
	for(int i = 0;i < N;i++)
	{
		if(!F[i])
		{
			ans.push_back(P[i]);
		}
	}
	long long inv = 0;
	BinaryIndexedTree<int> BIT(N);
	for(int i = 0;i < N;i++)
	{
		inv += BIT.sum(ans[i],N);
		BIT.add(ans[i],1);
	}
	cout << inv << "\n";
	for(int i = 0;i < N;i++)
	{
		cout << ++ans[i] << (i + 1 == N ? "\n" : " ");
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int tt = 1;
	cin >> tt;
	while(tt--) Main();
}
0