# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
658811 | aebov | T-Covering (eJOI19_covering) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#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;
}