#include<bits//stdc++.h> #include "aliens.h" using namespace std; using ll = long long; #define V vector const ll inf=1e18; long long take_photos(int n, int m, int k, std::vector<int> r, std::vector<int> c) { V<int>ind(n); for(int i=0;i<n;++i)ind[i]=i; auto cmp=[&](int i,int j){ if(min(r[i],c[i])==min(r[j],c[j]))return max(r[i],c[i])<max(r[j],c[j]); return min(r[i],c[i])<min(r[j],c[j]); }; sort(begin(ind),end(ind),cmp); ind.insert(begin(ind),n); r.push_back(0); c.push_back(0); V<V<ll>>dp(n+1,V<ll>(k+1,inf)); dp[0][0]=0; for(int it=1,ni,mx;it<=n;++it){ ni=ind[it]; for(int j=0;j<k;++j){ mx=0; for(int t=it,nt;t<=n;++t){ nt=ind[t]; mx=max({mx,r[nt],c[nt]}); dp[t][j+1]=min(dp[t][j+1],dp[it-1][j]+(ll)pow(mx-min(r[ni],c[ni])+1,2)); } } } return *min_element(begin(dp[n]),end(dp[n])); }

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