#919548 (C++17) No.2495 Three Sets

• 問題: No.2495 Three Sets
• ユーザー: cho435
• 提出日時: 2023-10-12 18:06:17
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 558 ms / 3,000 ms
• コード長: 1,007 bytes
• コンパイル時間: 4,386 ms
• コンパイル使用メモリ: 262,280 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-09-14 16:36:13
• 合計ジャッジ時間: 13,837 ms
• ジャッジサーバーID （参考情報）: judge4 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	294 ms; 5,248 KB
testcase_01	AC	307 ms; 5,248 KB
testcase_02	AC	431 ms; 5,376 KB
testcase_03	AC	300 ms; 5,376 KB
testcase_04	AC	298 ms; 5,376 KB
testcase_05	AC	297 ms; 5,376 KB
testcase_06	AC	323 ms; 5,376 KB
testcase_07	AC	469 ms; 5,376 KB
testcase_08	AC	456 ms; 5,376 KB
testcase_09	AC	301 ms; 5,376 KB
testcase_10	AC	300 ms; 5,376 KB
testcase_11	AC	479 ms; 5,376 KB
testcase_12	AC	479 ms; 5,376 KB
testcase_13	AC	508 ms; 5,376 KB
testcase_14	AC	533 ms; 5,376 KB
testcase_15	AC	515 ms; 5,376 KB
testcase_16	AC	558 ms; 5,376 KB
testcase_17	AC	558 ms; 5,376 KB
testcase_18	AC	111 ms; 5,376 KB
testcase_19	AC	196 ms; 5,376 KB
testcase_20	AC	554 ms; 5,376 KB

ソースコード

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using ll = long long;
#define rep(i,n) for(int i=0;i<(int)(n);i++)
using mint = atcoder::modint998244353;

int main(){
	int na,nb,nc;
	cin>>na>>nb>>nc;
	vector<int> a(na),b(nb),c(nc);
	rep(i,na) cin>>a.at(i);
	rep(i,nb) cin>>b.at(i);
	rep(i,nc) cin>>c.at(i);
	const ll shf=3005;
	vector<ll> ra(shf*2),rb(shf*2),rc(shf*2);
	rep(i,na) ra.at(a.at(i)+shf)++;
	rep(i,nb) rb.at(b.at(i)+shf)++;
	rep(i,nc) rc.at(c.at(i)+shf)++;
	vector<ll> sma(shf*2+1),csa(shf*2+1);
	for(int i=shf*2-1;i>=0;i--){
		sma.at(i)=sma.at(i+1)+ra.at(i)*(i-shf);
		csa.at(i)=csa.at(i+1)+ra.at(i);
	}
	ll ans=0;
	ll smb=0,bb=0;
	for(int lb=shf*2-1;lb>=0;lb--){
	  ll smc=0,cc=0;
		for(int lc=shf*2-1;lc>=0;lc--){
			int d;
			if(bb==0) d=-shf;
			else d=clamp(-smc/bb,-shf,shf);
			ll tmp=sma.at(d+shf)*bb+smb*cc+smc*csa.at(d+shf);
			ans=max(ans,tmp);
			smc+=rc.at(lc)*(lc-shf);
			cc+=rc.at(lc);
		}
		smb+=rb.at(lb)*(lb-shf);
		bb+=rb.at(lb);
	}
	cout<<ans<<endl;
}