#include <bits/stdc++.h>
using namespace std;
#define int int64_t
#define pb push_back
#define pll pair<ll ,ll>
#define F first
#define S second
#define pii pair<int ,int>
using namespace std;
const int N = (int)1e6 + 6;
int n, m, k, x, y, g[N], v, u, okt;
vector<int> vc, adj[N];
int hx[] = {-1, 1, 0, 0};
int hy[] = {0, 0, 1, -1};
bool vis[N], ok[N];
bool valid(int i,int j){return (i >= 0 && i < n && j >= 0 && j < m);}
void dfs(int v){
vis[v] = 1;
if(!ok[v])vc.pb(g[v]);
else okt ++;
for(auto u : adj[v])if(!vis[u])dfs(u);
}
int32_t main(){
ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
cin >> n >> m;
for(int i = 0; i < n; i ++)for(int j = 0; j < m;j ++)cin >> g[i*m + j], ok[i*m + j] = vis[i*m + j] = 0;
cin >> k;
ans = 0;
for(int i = 0; i < k; i ++){
cin >> x >> y;
int v = x*m + y;
// cout << "V : " << v << endl;
ok[v] = 1, ans += g[v];
for(int r = 0; r < 4; r ++){
int xx = x + hx[r], yy = y + hy[r];
if(valid(xx, yy) == 0)continue;
int u = xx * m + yy;
adj[v].pb(u), adj[u].pb(v);
// cout << " adj : " <<u << " " << v << endl;
}
}
for(int i = 0; i < n; i ++){
for(int j = 0; j < m; j ++){
v = i*m + j;
if(!vis[v] && ok[v]){
okt = 0;
vc.clear();
dfs(v);
if(vc.size() < 3 * okt)return cout << "NO\n", 0;
sort(vc.begin(), vc.end());
// cout << i << " " << j << " : " ;
// for(auto x : vc)cout << x << " ";
// cout << endl;
for(int f = 1; f <= 3*okt; f ++)ans += vc[vc.size() - f];
}
}
}return cout << ans << "\n" , 0;
}
Compilation message
covering.cpp: In function 'int32_t main()':
covering.cpp:30:2: error: 'ans' was not declared in this scope; did you mean 'abs'?
30 | ans = 0;
| ^~~
| abs
covering.cpp:51:18: warning: comparison of integer expressions of different signedness: 'std::vector<long int>::size_type' {aka 'long unsigned int'} and 'int64_t' {aka 'long int'} [-Wsign-compare]
51 | if(vc.size() < 3 * okt)return cout << "NO\n", 0;
| ~~~~~~~~~~^~~~~~~~~
