結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 1 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 169 ms
13,568 KB
testcase_05 AC 164 ms
13,824 KB
testcase_06 AC 165 ms
13,568 KB
testcase_07 AC 170 ms
13,568 KB
testcase_08 AC 164 ms
13,696 KB
testcase_09 AC 70 ms
32,632 KB
testcase_10 AC 70 ms
32,632 KB
testcase_11 AC 70 ms
32,636 KB
testcase_12 AC 69 ms
32,756 KB
testcase_13 AC 70 ms
32,760 KB
testcase_14 AC 2 ms
5,248 KB
testcase_15 AC 2 ms
5,248 KB
testcase_16 AC 2 ms
5,248 KB
testcase_17 AC 1 ms
5,248 KB
testcase_18 AC 2 ms
5,248 KB
testcase_19 AC 2 ms
5,248 KB
testcase_20 AC 1 ms
5,248 KB
testcase_21 AC 2 ms
5,248 KB
testcase_22 AC 2 ms
5,248 KB
testcase_23 AC 2 ms
5,248 KB
testcase_24 AC 1 ms
5,248 KB
testcase_25 AC 1 ms
5,248 KB
testcase_26 AC 2 ms
5,248 KB
testcase_27 AC 2 ms
5,248 KB
testcase_28 AC 2 ms
5,248 KB
testcase_29 AC 119 ms
10,880 KB
testcase_30 AC 9 ms
5,248 KB
testcase_31 AC 95 ms
9,728 KB
testcase_32 AC 56 ms
32,592 KB
testcase_33 AC 6 ms
5,248 KB
testcase_34 AC 56 ms
32,480 KB
testcase_35 AC 66 ms
32,556 KB
testcase_36 AC 6 ms
5,248 KB
testcase_37 AC 41 ms
6,400 KB
testcase_38 AC 172 ms
13,440 KB
testcase_39 AC 165 ms
13,440 KB
testcase_40 AC 97 ms
10,112 KB
testcase_41 AC 44 ms
6,912 KB
testcase_42 AC 162 ms
13,568 KB
testcase_43 AC 169 ms
13,696 KB
testcase_44 AC 53 ms
32,540 KB
testcase_45 AC 35 ms
18,228 KB
testcase_46 AC 30 ms
18,088 KB
testcase_47 AC 2 ms
5,248 KB
testcase_48 AC 2 ms
5,248 KB
testcase_49 AC 164 ms
13,568 KB
testcase_50 AC 164 ms
13,568 KB
testcase_51 AC 167 ms
13,696 KB
testcase_52 AC 152 ms
13,696 KB
testcase_53 AC 42 ms
6,016 KB
testcase_54 AC 148 ms
13,568 KB
testcase_55 AC 153 ms
13,696 KB
testcase_56 AC 159 ms
13,568 KB
testcase_57 AC 115 ms
13,688 KB
testcase_58 AC 125 ms
15,224 KB
testcase_59 AC 158 ms
11,136 KB
testcase_60 AC 2 ms
5,248 KB
testcase_61 AC 2 ms
5,248 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