結果

問題 No.919 You Are A Project Manager
ユーザー kotatsugamekotatsugame
提出日時 2019-10-26 00:20:29
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,924 bytes
コンパイル時間 1,148 ms
コンパイル使用メモリ 89,304 KB
実行使用メモリ 15,720 KB
最終ジャッジ日時 2024-09-13 11:20:09
合計ジャッジ時間 74,102 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 48 TLE * 7
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:36:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   36 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
//1-indexed
#include<vector>
template<typename T>
struct BIT{
	int n;
	vector<T>bit;
	BIT(int n_=0):n(n_),bit(n_+1){}
	T sum(int i)
	{
		T ans=0;
		for(;i>0;i-=i&-i)ans+=bit[i];
		return ans;
	}
	void add(int i,T a)
	{
		if(i==0)return;
		for(;i<=n;i+=i&-i)bit[i]+=a;
	}
	int lower_bound(T k)//k<=sum(ret)
	{
		if(k<=0)return 0;
		int ret=0,i=1;
		while((i<<1)<=n)i<<=1;
		for(;i;i>>=1)
			if(ret+i<=n&&bit[ret+i]<k)k-=bit[ret+=i];
		return ret+1;
	}
};
int N,A[10000];
vector<pair<int,int> >Q;
vector<pair<int,int> >Ai;
main()
{
	cin>>N;
	for(int i=0;i<N;i++)cin>>A[i];
	for(int i=0;i<N;i++)Ai.push_back(make_pair(A[i],i));
	for(int K=1;K<=N;K++)
	{
		for(int i=0;i+K<=N;i+=K)
		{
			Q.push_back(make_pair(i,i+K));
			Q.push_back(make_pair(N-i-K,N-i));
		}
	}
	vector<long>L(Q.size(),-1e9-1),R(Q.size(),1e9);
	sort(Ai.begin(),Ai.end());
	for(int c=0;c<100;c++)
	{
		vector<pair<long,int> >M;
		for(int i=0;i<Q.size();i++)
		{
			M.push_back(make_pair((L[i]+R[i])/2,i));
		}
		sort(M.begin(),M.end());
		int id=0;
		BIT<int>P(N);
		for(int i=0;i<=Ai.size();i++)
		{
			long T=i<Ai.size()?Ai[i].first:1e18;
			while(id<M.size()&&M[id].first<T)
			{
				int I=M[id].second;
				int l=Q[I].first,r=Q[I].second;
				bool flag=P.sum(r)-P.sum(l)>=(r-l+1)/2;
				if(flag)R[I]=M[id].first;
				else L[I]=M[id].first;
				id++;
			}
			if(i<Ai.size())P.add(Ai[i].second+1,1);
		}
	}
	vector<long>mid(R);
	long ans=0;
	int cnt=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]=mid[cnt++];
			R[i/K]=mid[cnt++];
		}
		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