Submission #1222939

#TimeUsernameProblemLanguageResultExecution timeMemory
1222939MalixAliens (IOI16_aliens)C++20
25 / 100
2098 ms125880 KiB
#include "aliens.h"
#include <bits/stdc++.h>
using namespace std;
 
typedef long long ll;
typedef vector<int> vi;
typedef vector<vi> vii;
typedef pair<int,int> pi;
typedef vector<pi> pii;
typedef tuple<int,int,int> ti;
typedef vector<ll> li;
typedef vector<li> lii;
 
#define REP(i,a,b) for(int i=a;i<b;i++)
#define F first
#define S second
#define PB push_back
#define LSOne(s) ((s)&(-s))
#define all(x) (x).begin(),(x).end()
 
ll INF=1000000000000000010;
int inf=1e9+10;
ll M=1e9+7;

ll dp[4000][4001];
vector<pair<ll,ll>> a;

ll dist(int p,int q){
    ll x=a[p].F,y=a[q].S;
    ll val=(y-x+1)*(y-x+1);
    if(p!=0){
        ll z=a[p-1].S-a[p].F+1;
        if(z>0)val-=z*z;
    }
    return val;
}

long long take_photos(int n, int m, int k, std::vector<int> r, std::vector<int> c) {
    REP(i,0,n)a.PB({max((ll)r[i],(ll)c[i]),min((ll)r[i],(ll)c[i])});
    sort(all(a));reverse(all(a));
    int pos=0;
    while(pos+1<a.size()){
        if(a[pos].S<=a[pos+1].S)a.erase(a.begin()+pos+1);
        else pos++;
    }
    pos=0;
    REP(i,0,n)swap(a[i].F,a[i].S);
    reverse(all(a));
    while(pos+1<a.size()){
        if(a[pos].S==a[pos+1].S&&a[pos].F<=a[pos+1].S)a.erase(a.begin()+pos+1);
        else pos++;
    }
    n=a.size();
    REP(i,0,4000)REP(j,0,4001)dp[i][j]=INF;
    REP(i,0,n)dp[i][1]=dist(0,i);
    REP(i,0,n)REP(j,2,k+1)REP(x,0,i)dp[i][j]=min(dp[i][j],dp[x][j-1]+dist(x+1,i));
    ll ans=INF;
    REP(i,1,k+1)ans=min(ans,dp[n-1][i]); 
    return ans;
}

Compilation message (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
      |         ^~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...