# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
1271955 | alexiah | Aliens (IOI16_aliens) | C++20 | 0 ms | 0 KiB |
#include "aliens.h"
#include <bits/stdc++.h>
using namespace std;
long long take_photos(int cant, int m, int k, vector<int> r, vector<int> c) {
vector<vector<int>> grid (r.size() , vector <int>(c.size()));
int n = r.size() , m = c.size();
for(int i = 0; i < r.size(); i++){
for(int j = 0; j < c.size(); j++){
grid[r[i]][c[i]]++;
}
}
for(int i = 0; i < r.size(); i++){
for(int j = 0; j < c.size(); j++){
grid[r[i]][c[i]]++;
}
}
int ans = 0;
for(int i = 0; i < n; i++){
int act = 0;
for(int j = 0; j < m; j++){
if(grid[i][j] > 0) act = max(act , j);
}
if(ans != -1)ans += abs((i+1)-(act+1)) * abs((i+1)-(act+1));
}
return ans;
}