結果

問題 No.919 You Are A Project Manager
ユーザー kotatsugame
提出日時 2019-10-25 23:57:43
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,081 ms / 3,000 ms
コード長 1,573 bytes
コンパイル時間 1,030 ms
コンパイル使用メモリ 84,316 KB
実行使用メモリ 7,168 KB
最終ジャッジ日時 2024-06-27 17:37:53
合計ジャッジ時間 25,651 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 55
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:47:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   47 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
#include<vector>
#include<algorithm>
template<typename T>
struct rangefreq{
	int n;
	vector<vector<T> >dat;
	rangefreq(const vector<T>&v={})
	{
		n=1;
		while(n<v.size())n<<=1;
		dat.assign(2*n-1,{});
		for(int i=0;i<v.size();i++)dat[i+n-1].push_back(v[i]);
		for(int i=n-2;i>=0;i--)
		{
			dat[i].resize(dat[i*2+1].size()+dat[i*2+2].size());
			merge(dat[i*2+1].begin(),dat[i*2+1].end(),
				dat[i*2+2].begin(),dat[i*2+2].end(),
				dat[i].begin()
			);
		}
	}
	int query(int a,int b,T x,int k=0,int l=0,int r=-1)const//[a,b) count(*<x)
	{
		if(r<0)r=n;
		if(b<=l||r<=a)return 0;
		else if(a<=l&&r<=b)return lower_bound(dat[k].begin(),dat[k].end(),x)-dat[k].begin();
		else return query(a,b,x,k*2+1,l,(l+r)/2)+query(a,b,x,k*2+2,(l+r)/2,r);
	}
};
int N;
int f(int L,int R,const rangefreq<int>&P)
{
	long l=-1e9,r=1e9+1;
	int K=R-L;
	while(r-l>1)
	{
		long m=(l+r)/2;
		if(P.query(L,R,m)>=(K+1)/2)r=m;
		else l=m;
	}
	return l;
}
main()
{
	cin>>N;
	vector<int>A(N);
	for(int&a:A)cin>>a;
	rangefreq<int>P(A);
	reverse(A.begin(),A.end());
	rangefreq<int>Q(A);
	long ans=0;
	for(int K=1;K<=N;K++)
	{
		int t=(K-1)/2;
		vector<long>L(N/K),R(N/K);
		for(int i=0;i+K<=N;i+=K)
		{
			L[i/K]=f(i,i+K,P);
			R[i/K]=f(i,i+K,Q);
		}
		for(int i=1;i<N/K;i++)
		{
			L[i]+=L[i-1];
			R[i]+=R[i-1];
		}
		for(int i=1;i<N/K;i++)
		{
			L[i]=max(L[i],L[i-1]);
			R[i]=max(R[i],R[i-1]);
		}
		for(int i=-1;i<N/K;i++)
		{
			int j=N/K-i-2;
			ans=max(ans,((i<0?0L:L[i])+(j<0?0L:R[j]))*K);
		}
	}
	cout<<ans<<endl;
}
0