Submission #471935

#	Time	Username	Problem	Language	Result	Execution time	Memory
471935		sofapuden	Aliens (IOI16_aliens)	C++14	4 / 100	1 ms	332 KiB

aliens

    #include "aliens.h"
    #include<bits/stdc++.h>
     
    using namespace std;
     
    typedef long long ll;
     
    const ll INF = (1ll<<60);
     
    bool cmp(pair<ll,ll> &a, pair<ll,ll> &b){
    	if(a.first == b.first)return a.second > b.second;
    	return a.first < b.first;
     
    }
     
    ll take_photos(int n, int m, int k, vector<int> r, vector<int> c) {
    	vector<pair<ll,ll>> v(n);
    	for(int i = 0; i < n; ++i){
    		v[i] = {r[i],c[i]};
    	}
    	for(int i = 0; i < n; ++i){
    		if(v[i].first > v[i].second)swap(v[i].first,v[i].second);
    	}
    	sort(v.begin(),v.end(),cmp);
    	ll l = 0, ri = INF, ans = INF, bes, cnt;
    	while(l <= ri){
    		ll m = (l+ri)>>1;
    		ll st = v[0].first, cu = v[0].second;
    		ll su = (cu-st+1)*(cu-st+1);
    		ll cn = 1;
    		for(int i = 0; i < n; ++i){
    			if(v[i].second <= cu)continue;
    			ll un = 0;
    			if(v[i].first <= cu){
    				un = (cu-v[i].first+1)*(cu-v[i].first+1);
    			}
    			if((v[i].second-v[i].first+1)*(v[i].second-v[i].first+1)-un+m < (v[i].second-st+1)*(v[i].second-st+1)-(cu-st+1)*(cu-st+1)){
    				cn++;
    				su+=(v[i].second-v[i].first+1)*(v[i].second-v[i].first+1)-un;
    				st = v[i].first;
    				cu = v[i].second;
    			}
    			else{
    				su+=(v[i].second-st+1)*(v[i].second-st+1)-(cu-st+1)*(cu-st+1);
    				cu = v[i].second;
    			}
    		}
    		if(cn < k){
    			cnt = cn;
    			bes = m;
    			ans = min(ans,su);
    			ri = m-1;
    		}
          	else if(cn == k){
              	cnt = cn;
              	bes = m;
              	ans = min(ans, su);
              	l = m+1;
            }
    		else{
    			l = m+1;
    		}
    	}
    	return ans;
    }

Compilation message (stderr)

aliens.cpp: In function 'll take_photos(int, int, int, std::vector<int>, std::vector<int>)':
aliens.cpp:25:37: warning: variable 'bes' set but not used [-Wunused-but-set-variable]
   25 |      ll l = 0, ri = INF, ans = INF, bes, cnt;
      |                                     ^~~
aliens.cpp:25:42: warning: variable 'cnt' set but not used [-Wunused-but-set-variable]
   25 |      ll l = 0, ri = INF, ans = INF, bes, cnt;
      |                                          ^~~

• Subtask 1: 4 / 4
• Subtask 2: 0 / 12
• Subtask 3: 0 / 9
• Subtask 4: 0 / 16
• Subtask 5: 0 / 19
• Subtask 6: 0 / 40