Submission #118135

#	Time	Username	Problem	Language	Result	Execution time	Memory
118135		scanhex	Naan (JOI19_naan)	C++17	100 / 100	1381 ms	149400 KiB

naan

#include <bits/stdc++.h>

using namespace std;
using nagai = __int128;
using big=__int128;
using unagai=unsigned long long;
using ll=long long;
#define sz(x) int((x).size())

const int N=2020;
int n,m;
nagai matr[N][N];
nagai pref[N][N];
nagai gcd(nagai a,nagai b){
	while(b){
		 a%=b,swap(a,b);
	}
	return a;
}
struct frac;
frac find(frac a,frac b);
const nagai C=1000000000LL;
struct frac{
	nagai a,b;
	frac(){}
	void find(){
		if(b<=C)return;
		frac r(a+1,b,0);
		int it=0;
		while(true){
			if(++it>40)break;
			frac nw=::find(*this,r);
//			cerr<<r.a<<' '<<r.b<<'\n';
			if(nw.a==a&&nw.b==b)return;
			if(nw.b>C){
				a=r.a,b=r.b;
				return;
			}
			r=nw;
		}
	}
	void check(){
		if(a==0)return;
		nagai g=gcd(a,b);
		a/=g;
		b/=g;
//		find();
	}
	frac(nagai a,nagai b,bool ch=false):a(a),b(b){
		if(ch)
			check();
	}
	frac(nagai x):a(x),b(1){}
	frac operator *(frac o){
		return frac(a*o.a,b*o.b);
	}
	frac operator +(frac o){
		return frac(a*o.b+o.a*b,b*o.b);
	}
	frac operator -(frac o){
		return frac(a*o.b-o.a*b,b*o.b);
	}
	bool operator <=(frac o){
		return a*o.b<=b*o.a;
	}
	bool operator <(frac o){
		return a*o.b<b*o.a;
	}
};
frac find(frac l, frac r){
	//cerr<<l.a<<' '<<l.b<<' '<<r.a<<' '<<r.b<<'\n';
	if(l.a>=l.b){
		nagai d=l.a/l.b;
		frac ans=find(frac(l.a-l.b*d,l.b,0),frac(r.a-r.b*d,r.b,0));
		ans.a+=ans.b*d;
		return ans;
	}
	if(r.a>r.b)
		return frac(1,1,0);
	frac ans=find(frac(r.b,r.a,0),frac(l.b,l.a,0));
	swap(ans.a,ans.b);
	return ans;
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin>>n>>m;
	for(int i=0;i<n;++i)
		for(int j=0;j<m;++j){
			ll x;
			cin>>x;
			matr[i][j]=x;
			pref[i][j+1]=pref[i][j]+matr[i][j];
		}
	frac cur(0);
	vector<frac>fr;
	vector<int>ord;
	vector<bool>used(n);
	for(int i=0;i<n;++i){
		frac mn(1e9);
		int mni=-1;
		for(int j=0;j<n;++j){
			if(used[j])continue;
//			frac kek=frac(matr[j][lb])*frac(cur.a%cur.b,cur.b);
			int L=0,R=m+1;
			while(R-L>1){
				int M=(L+R)/2;
				if(frac(pref[j][M])<=frac(pref[j][m])*frac(i+1,n))
					L=M;
				else
					R=M;
			}
			frac rest(0);
			rest=frac(pref[j][m])*frac(i+1,n)-frac(pref[j][L]);
			if(rest.a)
				rest=rest*frac(1,matr[j][L]);
			rest=rest+frac(L);
			rest.check();
			if(rest<mn)
				mn=rest,mni=j;
		}
		cur=mn;
//		cerr<<cur.a<<' '<<cur.b<<'\n';
		fr.push_back(cur);
		ord.push_back(mni);
		used[mni]=true;
	}
	for(int i=0;i<n-1;++i){
		cout<<(ll)fr[i].a<<' '<<(ll)fr[i].b<<'\n';
	}
	for(int i=0;i<n;++i)
		cout<<ord[i]+1<<' ';
	cout<<'\n';

	return 0;
}

• Subtask 1: 5 / 5
• Subtask 2: 24 / 24
• Subtask 3: 71 / 71