Submission #62332464

---

Source Code Expand

#include <bits/stdc++.h>
#define FOR(i, a, b) for (int i = a; i <= b; ++i)
#define ROF(i, a, b) for (int i = a; i >= b; --i)
#define all(x) x.begin(),x.end()
#define deSpace(x) cerr<<x<<' '
#define deEnter(x) cerr<<x<<'\n'

using namespace std;

typedef unsigned long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll> pll;
typedef array<ll,3> arr;
typedef double DB;
const ll MAXN=static_cast<ll>(2e5)+10,MAXval=static_cast<ll>(1e18)+3;

ll N,M,A[5][MAXN];
struct Triple{
	ll val,i,j,k;
};
struct cmp{
	bool operator()(const Triple &a,const Triple &b) const{
		return a.val<b.val;
	}
};
struct cmpSet{
	bool operator()(const Triple &a,const Triple &b) const{
		if(a.val!=b.val) return a.val<b.val;
		if(a.i!=b.i) return a.i<b.i;
		if(a.j!=b.j) return a.j<b.j;
		if(a.k!=b.k) return a.k<b.k;
		return false;
	}
};
priority_queue<Triple,vector<Triple> ,cmp> pq;
set<Triple,cmpSet> vis;

ll f(ll i,ll j,ll k){
	ll res=0;
	res=A[1][i]*A[2][j]+A[1][i]*A[3][k]+A[2][j]*A[3][k];
	return res;
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	cin>>N>>M;
	FOR(i, 1, 3){
		FOR(j,1,N){
			cin>>A[i][j];
		}
		sort(&A[i][1],&A[i][N+1],greater<ll>());
	}
	ll rank=0;
	pq.push((Triple){f(1,1,1),1,1,1});
	while(!pq.empty()){
		++rank;
		// deSpace(pq.top().val);deSpace(pq.top().i);deSpace(pq.top().j);deEnter(pq.top().k);
		if(rank==M){
			cout<<pq.top().val<<'\n';
			return 0;
		}
		ll i=pq.top().i,j=pq.top().j,k=pq.top().k;
		pq.pop();
		if(vis.find((Triple){f(i+1,j,k),i+1,j,k})==vis.end() && i+1<=N){
			pq.push((Triple){f(i+1,j,k),i+1,j,k});
			vis.insert((Triple){f(i+1,j,k),i+1,j,k});
		}
		if(vis.find((Triple){f(i,j+1,k),i,j+1,k})==vis.end() && j+1<=N){
			pq.push((Triple){f(i,j+1,k),i,j+1,k});
			vis.insert((Triple){f(i,j+1,k),i,j+1,k});
		}
		if(vis.find((Triple){f(i,j,k+1),i,j,k+1})==vis.end() && k+1<=N){
			pq.push((Triple){f(i,j,k+1),i,j,k+1});
			vis.insert((Triple){f(i,j,k+1),i,j,k+1});
		}
		
	}
	return 0;
}
/*

*/

Submission Info

Compile Error

./Main.cpp:50:3: warning: comparison of integers of different signs: 'int' and 'll' (aka 'unsigned long long') [-Wsign-compare]
                FOR(j,1,N){
                ^   ~   ~
./Main.cpp:2:40: note: expanded from macro 'FOR'
#define FOR(i, a, b) for (int i = a; i <= b; ++i)
                                     ~ ^  ~
./Main.cpp:15:39: warning: unused variable 'MAXval' [-Wunused-const-variable]
const ll MAXN=static_cast<ll>(2e5)+10,MAXval=static_cast<ll>(1e18)+3;
                                      ^
2 warnings generated.

Judge Result