제출 #1292008

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1292008		LmaoLmao	Aliens (IOI16_aliens)	C++20	0 / 100	2 ms	348 KiB

aliens

#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(ll delta) {
    deque<aa> q;
    q.push_back({v[1].fi*2,v[1].fi*v[1].fi+delta,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;
        }
    }
    return;
}

long long	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 dp[v.size()-1].fi-ans*k-skibidi;
    }
}

컴파일 시 표준 에러 (stderr) 메시지

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
      |         ^~~~

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