結果

問題 No.348 カゴメカゴメ
コンテスト
ユーザー kei
提出日時 2018-11-03 00:15:20
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 5,493 bytes
コンパイル時間 2,609 ms
コンパイル使用メモリ 203,160 KB
実行使用メモリ 89,336 KB
最終ジャッジ日時 2024-11-20 15:46:25
合計ジャッジ時間 8,240 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 23 WA * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
const int INF = 1e9;
const ll LINF = 1e18;
inline ll gcd(ll a, ll b) { return b ? gcd(b, a%b) : a; }
inline ll lcm(ll a, ll b) { return a / gcd(a, b)*b; }
template<class S,class T> ostream& operator << (ostream& out,const pair<S,T>& o){ out << "(" << o.first << "," << o.second << ")"; return out; }
template<class T> ostream& operator << (ostream& out,const vector<T> V){ for(int i = 0; i < V.size(); i++){ out << V[i]; if(i!=V.size()-1) out << " ";} return out; }
template<class T> ostream& operator << (ostream& out,const vector<vector<T> > Mat){ for(int i = 0; i < Mat.size(); i++) { if(i != 0) out << endl; out << Mat[i];} return out; }
template<class S,class T> ostream& operator << (ostream& out,const map<S,T> mp){ out << "{ "; for(auto it = mp.begin(); it != mp.end(); it++){ out << it->first << ":" << it->second; if(mp.size()-1 != distance(mp.begin(),it)) out << ", "; } out << " }"; return out; }

/*
 <url:https://yukicoder.me/problems/no/348>
 問題文============================================================
 =================================================================
 解説=============================================================
 ================================================================
 */

int dx[8] = {1,1,0,-1,-1,-1, 0, 1};
int dy[8] = {0,1,1,1 ,0 ,-1,-1,-1};
map<ll,ll> mp;
void rec(int n,vector<vector<int>>& T,vector<vector<ll>>& dp){
    ll sum1 = 0; // 使用しない
    ll sum2 = mp[n]; // 使用する
    for(auto next:T[n]){
        rec(next,T,dp);
        sum1 += max(dp[next][0],dp[next][1]);
        sum2 += dp[next][0];
    }
    dp[n][0] = sum1;
    dp[n][1] = sum2;
}

ll solve(){
    ll res = 0;
    int N,M; cin >> N >> M;
    vector<vector<char>> grid(N+2,vector<char>(M+2,'.'));
    for(int i = 1; i <= N; i++) for(int j = 1; j <= M;j++) cin >> grid[i][j];
    
    vector<vector<int>> checked(N+2,vector<int>(M+2,0));
    int dotcnt = 0;
    int ringcnt = 0;
    
    // 連なる.やxをグループ分け
    for(int i = 0; i < N+2; i++){
        for(int j = 0; j < M+2; j++){
            if(checked[i][j]!=0) continue;
            char c = grid[i][j];
            int add = 0;
            if(c == '.'){ dotcnt--; add = 2;}
            else{ ringcnt++; add = 1;}
            queue<pii> q; q.push({i,j});
            while(q.size()){
                int x,y;
                tie(y,x) = q.front(); q.pop();
                for(int k = 0; k < 8; k+=add){
                    int nx = x + dx[k];
                    int ny = y + dy[k];
                    if(ny < 0 || nx < 0 || ny >= N+2 || nx >= M+2) continue;
                    if(checked[ny][nx]!=0) continue;
                    if(grid[ny][nx] != c) continue;
                    if(c == '.'){
                        checked[ny][nx] = dotcnt;
                        q.push({ny,nx});
                    }else{
                        checked[ny][nx] = ringcnt;
                        q.push({ny,nx});
                    }
                }
            }
        }
    }
    
    // グループ分けしたものからグラフを作る
    int SZ = abs(dotcnt)+1+abs(ringcnt);
    vector<vector<pii>> G(SZ);
    for(int i = 0; i < N+2; i++){
        for(int j = 0; j < M+2; j++){
            char c = grid[i][j];
            int add = 0;
            if(c == '.'){ add = 2;}
            else{ add = 1;}
            
            pii S;
            if(checked[i][j] < 0) S = {abs(checked[i][j]),0};
            else S = {abs(dotcnt) + checked[i][j],1};
            
            for(int k = 0; k < 8; k+=add){
                int nx = j + dx[k];
                int ny = i + dy[k];
                if(ny < 0 || nx < 0 || ny >= N+2 || nx >= M+2) continue;
                if(checked[i][j] == checked[ny][nx]) continue;
                pii T;
                if(checked[ny][nx] < 0) T = {abs(checked[ny][nx]),0};
                else T = {abs(dotcnt) + checked[ny][nx],1};
                G[S.first].push_back(T);
                G[T.first].push_back(S);
            }
        }
    }
    for(int i = 0; i < SZ;i++){
        sort(G[i].begin(),G[i].end());
        G[i].erase(unique(G[i].begin(),G[i].end()),G[i].end());
    }
    
    // リングの大きさを数える
    for(int i = 0; i < N+2; i++){
        for(int j = 0; j < M+2; j++){
            if(checked[i][j] > 0){
                mp[checked[i][j]]++;
            }
        }
    }
    
    // 外側のリングから内側のリングへグラフを貼って木を作る 0 はコスト0の仮想のリング
    vector<vector<int>> T(abs(ringcnt)+1);
    vector<int> flag(abs(ringcnt)+1);
    for(auto p:G[1]){ T[0].push_back(p.first-abs(dotcnt)); flag[p.first-abs(dotcnt)] = 1;}
    for(int i = 2; i <= abs(dotcnt);i++){
        for(auto s:G[i]){
            if(flag[s.first-abs(dotcnt)]==0) continue;
            for(auto t:G[i]){
                if(s.first == t.first) continue;
                T[s.first-abs(dotcnt)].push_back(t.first-abs(dotcnt));
                flag[t.first-abs(dotcnt)] = 1;
            }
            break;
        }
    }
    
    // dp[i][フラグ] := 頂点iを見たとき、自分を使用したかどうか？
    vector<vector<ll>> dp(abs(ringcnt)+1,vector<ll>(2));
    rec(0,T,dp);
    res = max(dp[0][0],dp[0][1]);
    return res;
}
int main(void) {
    cin.tie(0); ios_base::sync_with_stdio(false);
    cout << solve() << endl;
    return 0;
}
0