Submission #1291949

#TimeUsernameProblemLanguageResultExecution timeMemory
1291949LmaoLmaoAliens (IOI16_aliens)C++20
Compilation error
0 ms0 KiB
#include "aliens.h"
#include <bits/stdc++.h>
#define fi first
#define se second
#define name "a"
using namespace std;

using ii = pair<long long,long long>;
using aa = array<long long,3>;
using ll = long long;
const int N=1e6+5;
const int MOD=1e9+7;

ii dp[N];
int a[N];
int b[N];
int m,n;
vector<ii> v;

bool cmp(aa a,aa b,long long cur) {
    if(a[0]*cur+a[1]!=b[0]*cur+b[1]) {
        return a[0]*cur+a[1]<b[0]*cur+b[1];
    } else
    return a[2]<b[2];
}


void alien(long long delta) {
    deque<aa> q;
    q.push_back({v[1].fi*2,0,0});
    for(int i=1;i<v.size();i++) {
        while(q.size()>1 && !cmp(q[0],q[1],v[i].se)) {
            q.pop_front();
        }
        dp[i].fi=q[0][0]*v[i].se+q[0][1]+v[i].se*v[i].se;
        dp[i].se=q[0][2]+1;
        if(i<v.size()-1) {
            aa cur={v[i+1].fi*2,dp[i].fi+v[i+1].fi*v[i+1].fi+delta,dp[i].se};
            q.push_back(cur);
            //cout << cur[0] << ' ' << cur[1] << endl;
        }
        //cout << dp[i].fi << endl;
    }
    return;
}

int	take_photos(int	M,	int	N,	int	k,	vector<int> R,	vector<int> C) {
    m=M;
    n=N;
    for(int i=0;i<m;i++) {
        int x=R[i+1],y=C[i+1];
        if(x<y) {
            a[y]=max(a[y],y-x+1);
        }
        else {
            a[x]=max(a[x],x-y+1);
        }
    }
    long long tmp=-1e18;
    for(int i=n;i>0;i--) {
        if(a[i]-i<=tmp) a[i]=0;
        tmp=max(tmp,(ll)a[i]-i);
        b[i]=a[i];
    }
    v.push_back({0,0});
    for(int i=1;i<=n;i++) {
        if(a[i]) v.push_back({a[i]-i,i});
    }
    long long l=0,r=n*n;
    long long ans=0;
    while(l<=r) {
        long long mid = l + r >> 1;
        alien(mid);
        if(dp[v.size()-1].se<=k) {
            ans=mid;
            r=mid-1;
        }
        else {
            l=mid+1;
        }
    }
    ll skibidi=0;
    for(int i=1;i<v.size();i++) {
        if(v[i].se-a[v[i].se]<v[i-1].se) {
            int tmp=v[i-1].se-(v[i].se-a[v[i].se]);
            skibidi+=tmp*tmp;
        }
    }
    alien(ans);
    if(ans==0) {
        return dp[v.size()-1].fi-skibidi;
    }
    else {
        return cout << dp[v.size()-1].fi+ans*k-skibidi;
    }
    return 0;
}

Compilation message (stderr)

aliens.cpp:47:9: error: ambiguating new declaration of 'int take_photos(int, int, int, std::vector<int>, std::vector<int>)'
   47 | int     take_photos(int M,      int     N,      int     k,      vector<int> R,  vector<int> C) {
      |         ^~~~~~~~~~~
In file included from aliens.cpp:1:
aliens.h:5:11: note: old declaration 'long long int take_photos(int, int, int, std::vector<int>, std::vector<int>)'
    5 | long long take_photos(int n, int m, int k, std::vector<int> r, std::vector<int> c);
      |           ^~~~~~~~~~~
aliens.cpp: In function 'int take_photos(int, int, int, std::vector<int>, std::vector<int>)':
aliens.cpp:94:21: error: cannot convert 'std::basic_ostream<char>::__ostream_type' {aka 'std::basic_ostream<char>'} to 'int' in return
   94 |         return cout << dp[v.size()-1].fi+ans*k-skibidi;
      |                ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
      |                     |
      |                     std::basic_ostream<char>::__ostream_type {aka std::basic_ostream<char>}
aliens.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~
aliens_c.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~