Submission #524414

#	Time	Username	Problem	Language	Result	Execution time	Memory
524414		Koosha_mv	Naan (JOI19_naan)	C++14	100 / 100	622 ms	128740 KiB

naan

#include <bits/stdc++.h>
using namespace std;
#define dbgv(v) cout<<#v<<" = "; f(i,0,v.size()) cout<<v[i]<<" "; cout<<endl
#define dbga(a,x,y) cout<<#a<<" = "; f(i,x,y) cout<<a[i]<<" "; cout<<endl
#define erorp(x) cout<<#x<<"={"<<(x.F)<<" , "<<x.S<<"}"<<endl
#define eror(x) cout<<#x<<'='<<(x)<<endl
#define f_(i,a,b) for(int i=a;i>=b;i--)
#define f(i,a,b) for(int i=a;i<b;i++)
#define nb(x) __builtin_popcount(x)
#define all(v) v.begin(),v.end()
#define bit(n,k) (((n)>>(k))&1)
#define Add(x,y) x=(x+y)%mod
#define maxm(a,b) a=max(a,b)
#define minm(a,b) a=min(a,b)
#define lst(x) x[x.size()-1]
#define sz(x) int(x.size())
#define mp make_pair
#define ll long long
#define pb push_back
#define S second
#define F first
#define int ll

const int N=2020;

int n,m,a[N][N],sum[N],mark[N];
vector<int> p;
vector<pair<int,int> > ans;
vector<pair<int,pair<int,int> > > v[N];

bool cmp(pair<int,pair<int,int> > i,pair<int,pair<int,int> > j){
	pair<__int128,__int128> u=i.S,v=j.S;
	return u.F*v.S<u.S*v.F;
}
void solve(int x){
	int p=1;
	f(i,1,n+1){
		while(a[x][p]<i*sum[x]){
			p++;
		}
		//cout<<p<<" "<<a[x][p]<<" "<<sum[i]<<endl;
		pair<int,int> e;
		e.S=a[x][p]-a[x][p-1];
		e.F=e.S*(p-1)+(i*sum[x])-a[x][p-1];		
		v[i].pb(mp(x,e));
		//cout<<x<<" "<<i<<" -> "<<e.F<<" "<<e.S<<endl;
	}	
}

main(){
	ios::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);
	cin>>n>>m;
	f(i,1,n+1){
		f(j,1,m+1){
			cin>>a[i][j];
			sum[i]+=a[i][j];
			a[i][j]*=n;
			a[i][j]+=a[i][j-1];		
		}
	}
	f(i,1,n+1){
		solve(i);
	}
	f(i,1,n+1){
		sort(all(v[i]),cmp);
		for(auto x : v[i]){
			if(!mark[x.F]){
				mark[x.F]=1;
				p.pb(x.F);
				ans.pb(x.S);
				break;
			}
		}
	}
	f(i,0,ans.size()-1){
		cout<<ans[i].F<<" "<<ans[i].S<<'\n';
	}
	f(i,0,p.size()) cout<<p[i]<<" ";
}

Compilation message (stderr)

naan.cpp:50:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   50 | main(){
      | ^~~~
naan.cpp: In function 'int main()':
naan.cpp:8:31: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    8 | #define f(i,a,b) for(int i=a;i<b;i++)
......
   75 |  f(i,0,ans.size()-1){
      |    ~~~~~~~~~~~~~~~~            
naan.cpp:75:2: note: in expansion of macro 'f'
   75 |  f(i,0,ans.size()-1){
      |  ^
naan.cpp:8:31: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    8 | #define f(i,a,b) for(int i=a;i<b;i++)
......
   78 |  f(i,0,p.size()) cout<<p[i]<<" ";
      |    ~~~~~~~~~~~~                
naan.cpp:78:2: note: in expansion of macro 'f'
   78 |  f(i,0,p.size()) cout<<p[i]<<" ";
      |  ^

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