結果

問題 No.1060 素敵な宝箱
ユーザー kotatsugame
提出日時 2020-05-22 22:04:54
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 28 ms / 2,000 ms
コード長 841 bytes
コンパイル時間 709 ms
コンパイル使用メモリ 77,656 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-05 17:35:26
合計ジャッジ時間 1,790 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:9:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
    9 | main()
      | ^~~~

ソースコード

diff #

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int N,M;
int A[300][300];
long C[300],D[300];
bool vis[300];
main()
{
	cin>>N>>M;
	for(int i=0;i<N;i++)
	{
		for(int j=0;j<M;j++)
		{
			cin>>A[i][j];
			C[j]+=A[i][j];
		}
	}
	for(int _=1;_<N;_+=2)
	{
		long X=0;
		for(int j=0;j<M;j++)
		{
			X+=C[j]*C[j];
			X-=D[j]*D[j];
		}
		vector<pair<long,int> >Y;
		for(int i=0;i<N;i++)
		{
			if(vis[i])continue;
			long nX=0;
			for(int j=0;j<M;j++)
			{
				nX+=(C[j]-A[i][j])*(C[j]-A[i][j]);
				nX-=(D[j]+A[i][j])*(D[j]+A[i][j]);
			}
			Y.push_back(make_pair(nX-X,i));
		}
		sort(Y.begin(),Y.end());
		int id=Y[1].second;
		vis[Y[0].second]=true;
		vis[id]=true;
		for(int j=0;j<M;j++)
		{
			C[j]-=A[id][j];
			D[j]+=A[id][j];
		}
	}
	long ans=0;
	for(int j=0;j<M;j++)ans+=C[j]*C[j]-D[j]*D[j];
	cout<<ans<<endl;
}
0