結果

問題 No.2304 Distinct Elements
ユーザー kotatsugamekotatsugame
提出日時 2023-05-12 22:47:23
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 192 ms / 3,000 ms
コード長 1,609 bytes
コンパイル時間 1,212 ms
コンパイル使用メモリ 91,972 KB
実行使用メモリ 32,760 KB
最終ジャッジ日時 2024-05-06 12:50:11
合計ジャッジ時間 8,046 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 183 ms
13,696 KB
testcase_05 AC 184 ms
13,696 KB
testcase_06 AC 183 ms
13,568 KB
testcase_07 AC 182 ms
13,696 KB
testcase_08 AC 183 ms
13,568 KB
testcase_09 AC 93 ms
32,760 KB
testcase_10 AC 93 ms
32,588 KB
testcase_11 AC 93 ms
32,636 KB
testcase_12 AC 93 ms
32,632 KB
testcase_13 AC 92 ms
32,760 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 3 ms
5,376 KB
testcase_18 AC 2 ms
5,376 KB
testcase_19 AC 2 ms
5,376 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 2 ms
5,376 KB
testcase_23 AC 2 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 2 ms
5,376 KB
testcase_29 AC 135 ms
10,880 KB
testcase_30 AC 9 ms
5,376 KB
testcase_31 AC 106 ms
9,600 KB
testcase_32 AC 76 ms
32,464 KB
testcase_33 AC 7 ms
5,376 KB
testcase_34 AC 78 ms
32,480 KB
testcase_35 AC 85 ms
32,680 KB
testcase_36 AC 6 ms
5,376 KB
testcase_37 AC 44 ms
6,528 KB
testcase_38 AC 183 ms
13,440 KB
testcase_39 AC 186 ms
13,568 KB
testcase_40 AC 110 ms
10,240 KB
testcase_41 AC 50 ms
6,912 KB
testcase_42 AC 178 ms
13,568 KB
testcase_43 AC 192 ms
13,696 KB
testcase_44 AC 72 ms
32,408 KB
testcase_45 AC 48 ms
18,096 KB
testcase_46 AC 39 ms
17,960 KB
testcase_47 AC 2 ms
5,376 KB
testcase_48 AC 2 ms
5,376 KB
testcase_49 AC 189 ms
13,696 KB
testcase_50 AC 189 ms
13,696 KB
testcase_51 AC 188 ms
13,696 KB
testcase_52 AC 165 ms
13,568 KB
testcase_53 AC 47 ms
6,016 KB
testcase_54 AC 170 ms
13,568 KB
testcase_55 AC 172 ms
13,696 KB
testcase_56 AC 175 ms
13,568 KB
testcase_57 AC 131 ms
13,688 KB
testcase_58 AC 141 ms
15,356 KB
testcase_59 AC 174 ms
11,136 KB
testcase_60 AC 2 ms
5,376 KB
testcase_61 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<algorithm>
#include<map>
#include<vector>
#include<cassert>
using namespace std;
struct dat{
	long long Lc,Rc,Ls,Rs;
	map<int,int>cnt;
	int val;
	dat(){}
	dat(int v):Lc(0LL),Rc(0LL),Ls(0LL),Rs(0LL),val(v)
	{
		cnt.clear();
		cnt[v]=1;
	}
	long long need()
	{
		return Rs-Rc*val+Lc*val-Ls;
	}
	bool dec()
	{
		long long now=need();
		long long nLc=Lc,nRc=Rc,nLs=Ls,nRs=Rs;
		if(cnt.find(val)!=cnt.end())
		{
			int c=cnt[val];
			nRc+=c;
			nRs+=(long long)c*val;
		}
		if(cnt.find(val-1)!=cnt.end())
		{
			int c=cnt[val-1];
			nLc-=c;
			nLs-=(long long)c*(val-1);
		}
		long long nxt=nRs-nRc*(val-1)+nLc*(val-1)-nLs;
		if(nxt<=now)
		{
			val--;
			Lc=nLc;
			Rc=nRc;
			Ls=nLs;
			Rs=nRs;
			return true;
		}
		else return false;
	}
};
void merge(dat&l,dat&r)
{
	assert(l.val==r.val);
	l.Lc+=r.Lc;
	l.Rc+=r.Rc;
	l.Ls+=r.Ls;
	l.Rs+=r.Rs;
	if(l.cnt.size()<r.cnt.size())l.cnt.swap(r.cnt);
	for(pair<int,int>p:r.cnt)l.cnt[p.first]+=p.second;
}
int N,A[2<<17];
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cin>>N;
	for(int i=0;i<N;i++)cin>>A[i];
	sort(A,A+N);
	for(int i=0;i<N;i++)A[i]-=i;
	vector<dat>S;
	S.push_back(dat(A[0]));
	for(int i=1;i<N;i++)
	{
		if(S.back().val<A[i])
		{
			S.push_back(dat(A[i]));
		}
		else
		{
			int v=S.back().val;
			if(v>A[i])S.back().Lc++,S.back().Ls+=A[i];
			S.back().cnt[A[i]]++;
			while(S.back().dec())
			{
				if(S.size()>=2&&S[S.size()-2].val==S.back().val)
				{
					merge(S[S.size()-2],S.back());
					S.pop_back();
				}
			}
		}
	}
	long long ans=0;
	for(int i=0;i<S.size();i++)ans+=S[i].need();
	cout<<ans<<endl;
}
0