제출 #756197

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
756197		alexander707070	Aliens (IOI16_aliens)	C++14	0 / 100	1 ms	340 KiB

aliens

#include <bits/stdc++.h>
using namespace std;

const long long inf=1e16;

struct seg{
    long long l,r;

    inline friend bool operator < (seg fr,seg sc){
        if(fr.l!=sc.l)return fr.l<sc.l;
        return fr.r<sc.r;
    }
};

int n,m,k;
seg a[50007];
vector<seg> v;

long long dp[50007][4007];

long long cost(int l,int r){
    if(l==1 or v[l].l>v[l-1].r)return (v[r].r-v[l].l+1)*(v[r].r-v[l].l+1); 
    return (v[r].r-v[l].l+1)*(v[r].r-v[l].l+1)-(v[l-1].r-v[l].l+1)*(v[l-1].r-v[l].l+1);
}

void solve(int l,int r,int optl,int optr,int k){
    if(l>r)return;

    int mid=(l+r)/2,opt;
    dp[mid][k]=inf;

    for(int i=min(mid,optr);i>=optl;i--){
        if(dp[mid-1][k-1]+cost(i,mid)){
            dp[mid][k]=min(dp[mid][k],dp[i-1][k-1]+cost(i,mid));
            opt=i;
        }
    }

    solve(l,mid-1,optl,opt,k);
    solve(mid+1,r,opt,optr,k);
}

long long take_photos(int N, int M, int K, vector<int> r, vector<int> c){
    n=N; m=M; k=K;

    for(int i=0;i<n;i++){
        a[i+1]={min(r[i],c[i]),max(r[i],c[i])};
    }
    sort(a+1,a+n+1);

    v.push_back({-1,-1});
    for(int i=1;i<=n;i++){
        while(!v.empty() and v.back().l>=a[i].l and v.back().r<=a[i].r)v.pop_back();
        if(v.back().l<=a[i].l and v.back().r>=a[i].r)continue;
        v.push_back(a[i]);
    }

    dp[0][0]=0;
    for(int i=1;i<v.size();i++)dp[i][0]=inf;
    
    for(int i=1;i<=k;i++){
        solve(1,v.size()-1,1,v.size()-1,i);
    }

    return dp[v.size()-1][k];
}

/*
int main(){

    cout<<take_photos(5, 7, 2, {0, 4, 4, 4, 4}, {3, 4, 6, 5, 6})<<"\n";

}
*/

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

aliens.cpp: In function 'long long int take_photos(int, int, int, std::vector<int>, std::vector<int>)':
aliens.cpp:59:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<seg>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   59 |     for(int i=1;i<v.size();i++)dp[i][0]=inf;
      |                 ~^~~~~~~~~
aliens.cpp: In function 'void solve(int, int, int, int, int)':
aliens.cpp:39:10: warning: 'opt' may be used uninitialized in this function [-Wmaybe-uninitialized]
   39 |     solve(l,mid-1,optl,opt,k);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~

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