結果

問題 No.1324 Approximate the Matrix
ユーザー soto800soto800
提出日時 2020-12-21 23:14:19
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 3,227 bytes
コンパイル時間 1,683 ms
コンパイル使用メモリ 174,180 KB
実行使用メモリ 10,752 KB
最終ジャッジ日時 2024-09-21 13:13:25
合計ジャッジ時間 7,150 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 471 ms
10,752 KB
testcase_01 AC 470 ms
5,376 KB
testcase_02 AC 583 ms
5,376 KB
testcase_03 TLE -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
testcase_42 -- -
testcase_43 -- -
testcase_44 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#define lli long long int
#define REP(i,s,n) for(lli i=s;i<n;i++)
#define NUM 2520
#define INF (1LL<<50)
#define DEBUG 0
#define mp(a,b) make_pair(a,b)
#define SORT(V) sort(V.begin(),V.end())
#define PI (3.141592653589794)
#define MOD (1000000007)
#define TO_STRING(VariableName) # VariableName
#define LOG(xx) if(DEBUG)cout<<TO_STRING(xx)<<"="<<xx<<" "<<endl;
#define LOG2(xx,yy) if(DEBUG)cout<<TO_STRING(xx)<<"="<<xx<<" "<<TO_STRING(yy)<<"="<<yy<<endl;
#define LOG3(xx,yy,z) if(DEBUG)cout<<TO_STRING(xx)<<"="<<xx<<" "<<TO_STRING(yy)<<"="<<yy<<" "<<TO_STRING(z)<<"="<<z<<endl;
#define LOG4(w,xx,yy,z) if(DEBUG)cout<<TO_STRING(w)<<"="<<w<<" "<<TO_STRING(xx)<<"="<<xx<<" "<<TO_STRING(yy)<<"="<<yy<<" "<<TO_STRING(z)<<"="<<z<<endl;

template<class T>bool chmax(T & a, const T & b) { if (a < b) { a = b; return 1; } return 0; }
template<class T>bool chmin(T& a, const T& b) { if (b < a) { a = b; return 1; } return 0; }

lli p[210][210];
lli q[210][210];

lli pow2(lli val){
	return val*val;
}

void solve(){

	lli N,K;
	cin>>N>>K;
	vector<lli> a(N),b(N);
	for(auto &e:a)cin>>e;
	for(auto &e:b)cin>>e;

	lli maxA = -1,maxAI = -1;
	lli maxB = -1,maxBI = -1;

	REP(i,0,N){
		REP(j,0,N){
			cin>>p[i][j];
		}
	}

	REP(i,0,N){
		if(maxA < a[i]){
			maxA = a[i];
			maxAI = i;
		}		
		if(maxB < b[i]){
			maxB = b[i];
			maxBI = i;
		}
	}

	REP(i,0,N){
		REP(j,0,N){
			if(i == maxAI && j == maxBI)continue;
			lli now = min(a[i],b[j]);
			q[i][j] = now;
			a[i]-=now;
			b[j]-=now;
		}
	}

	lli subA = 0,subB = 0;
	REP(i,0,N){
		subA += a[i];
		subB += b[i];
	}
	//LOG2(subA,subB);

	q[maxAI][maxBI] = subA;
/*
	//if(DEBUG){
		REP(i,0,N){
			REP(j,0,N){
				cout<<q[i][j]<<" ";
			}
			cout<<endl;
		}
	//}*/

	lli nowScore = 0;
	REP(i,0,N){
		REP(j,0,N){
			LOG4(i,j,p[i][j],q[i][j]);
			nowScore += pow2(p[i][j]-q[i][j]);
		}
	}
	LOG(nowScore);

	std::random_device rnd;     // 非決定的な乱数生成器を生成
    std::mt19937 mt(rnd());     //  メルセンヌ・ツイスタの32ビット版、引数は初期シード値
    std::uniform_int_distribution<> rand(0, N-1);        // [0, 99] 範囲の一様乱数
	lli cnt = 0;

	lli id[2]={0,0};
	lli jd[2]={0,0};

	while(cnt < 1000000){

		lli fromI = rand(mt);
		lli toI   = rand(mt);
		lli fromJ = rand(mt);
		lli toJ   = rand(mt);

		if(q[fromI][fromJ] <= 0 || q[toI][toJ] <= 0)continue;
		if(fromI == toI || fromJ == toJ) continue;

		id[0]=fromI,id[1]=toI;
		jd[0]=fromJ,jd[1]=toJ;

		//LOG4(fromI,toI,fromJ,toJ);

		lli nexScore = nowScore;
		REP(i,0,2)REP(j,0,2)nexScore -= pow2(p[id[i]][jd[j]]-q[id[i]][jd[j]]);

		//LOG(nexScore);
		q[fromI][fromJ]--;
		q[fromI][toJ]++;
		q[toI][toJ]--;
		q[toI][fromJ]++;
		REP(i,0,2)REP(j,0,2)nexScore += pow2(p[id[i]][jd[j]]-q[id[i]][jd[j]]);

		if(0){
			LOG2(cnt,nexScore);
			REP(i,0,N){
				REP(j,0,N){
					cout<<q[i][j]<<" ";
				}
				cout<<endl;
			}
		}
		if(nexScore > nowScore){
			q[fromI][fromJ]++;
			q[fromI][toJ]--;
			q[toI][toJ]++;
			q[toI][fromJ]--;
		}
		else{
			nowScore = nexScore;
		}
		cnt++;
	}

	cout<<nowScore<<endl;

}

int main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);

    lli t=1;
    //cin>>t;
    while(t--)solve();

    return 0;
}
0