結果

問題 No.5016 Worst Mayor
コンテスト
ユーザー FplusFplusF
提出日時 2023-04-29 15:40:04
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,215 ms / 2,000 ms
コード長 6,114 bytes
コンパイル時間 3,534 ms
コンパイル使用メモリ 244,464 KB
実行使用メモリ 206,152 KB
スコア 18,691,029,076
平均クエリ数 400.00
最終ジャッジ日時 2023-04-29 15:41:14
合計ジャッジ時間 60,489 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 50
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
const int INF=1e9;
using pii=pair<int,int>;
using tii=tuple<int,int,int>;
#define rep(i,n) for (int i=0;i<(int)(n);i++)
#define all(v) v.begin(),v.end()
template<class T> void chmin(T &a,T b){
    if(a>b){
        a=b;
    }
}
template<class T> void chmax(T &a,T b){
    if(a<b){
        a=b;
    }
}
vector<int> pi={0,1,0,-1},pj={1,0,-1,0};
int to(int i,int j){
    return i*14+j;
}
pii back(int x){
    return {x/14,x%14};
}
struct input{
    int n,t;
    vector<int> s,g;
    void input_cin(){
        cin >> n >> t;
        s.resize(n);
        g.resize(n);
        rep(i,n){
            int a,b,c,d;
            cin >> a >> b >> c >> d;
            a--;
            b--;
            c--;
            d--;
            s[i]=to(a,b);
            g[i]=to(c,d);
        }
    }
}input;
struct solver{
    vector<vector<int>> people;
    pair<vector<vector<int>>,int> dijkstra(vector<vector<pii>> &g){
        int score=0;
        vector<vector<int>> root_cnt(14*14,vector<int>(14*14,0));
        rep(i,14*14){
            vector<int> dist(14*14,INF);
            vector<int> prev(14*14);
            priority_queue<pii> pq;
            dist[i]=0;
            pq.push({i,i});
            while(!pq.empty()){
                int x,d;
                tie(x,d)=pq.top();
                pq.pop();
                for(auto &[to,w]:g[x]){
                    if(dist[x]+w<dist[to]){
                        dist[to]=dist[x]+w;
                        prev[to]=x;
                        pq.push({to,dist[x]+w});
                    }
                }
            }
            rep(j,14*14){
                if(people[i][j]==0) continue;
                int now=j,cnt=0;
                while(now!=i){
                    if(dist[now]-dist[prev[now]]==223) cnt++;
                    root_cnt[prev[now]][now]++;
                    now=prev[now];
                }
                score+=60*cnt*people[i][j];
            }
        }
        return {root_cnt,score};
    }
    void solve(){
        vector<vector<pii>> g(14*14);
        rep(i,14){
            rep(j,14){
                rep(k,4){
                    int ti=i+pi[k],tj=j+pj[k];
                    if(ti<0||14<=ti||tj<0||14<=tj) continue;
                    g[to(i,j)].emplace_back(to(ti,tj),1000);
                }
            }
        }
        people.resize(14*14);
        rep(i,14*14) people[i].resize(14*14,0);
        rep(i,input.n){
            people[input.s[i]][input.g[i]]++;
        }
        vector<int> score_sum={0};
        vector<pii> v;
        vector<vector<bool>> built(14*14,vector<bool>(14*14,false));
        int ns=-1,ng=-1;
        vector<vector<int>> cnt=dijkstra(g).first;
        rep(r,70){;
            int max_cnt=0,ts=-1,tg=-1;
            rep(i,14*14){
                rep(j,14*14){
                    if(cnt[i][j]==0) continue;
                    if(built[i][j]) continue;
                    if(max_cnt<cnt[i][j]){
                        max_cnt=cnt[i][j];
                        ts=i;
                        tg=j;
                    }
                }
            }
            if(ts==-1||tg==-1) break;
            for(auto &[to,w]:g[ts]){
                if(to==tg) w=223;
            }
            for(auto &[to,w]:g[tg]){
                if(to==ts) w=223;
            }
            int now_score;
            tie(cnt,now_score)=dijkstra(g);
            score_sum.push_back(now_score);
            ns=ts;
            ng=tg;
            built[ns][ng]=true;
            built[ng][ns]=true;
            v.emplace_back(ns,ng);
        }
        int x=v.size();
        vector<vector<vector<int>>> dp(input.t+1,vector<vector<int>>(input.t+1,vector<int>(x,-INF)));
        vector<vector<vector<tii>>> prev(input.t+1,vector<vector<tii>>(input.t+1,vector<tii>(x)));
        dp[0][1][0]=1e6;
        rep(i,input.t){
            rep(j,input.t+1){
                rep(k,x){
                    if(dp[i][j][k]==-INF) continue;
                    if(0<j&&k+1<x){
                        int cost=1e7/sqrt(j);
                        if(cost<=dp[i][j][k]&&dp[i+1][j][k+1]<dp[i][j][k]-cost+score_sum[k+1]){
                            dp[i+1][j][k+1]=dp[i][j][k]-cost+score_sum[k];
                            prev[i+1][j][k+1]={i,j,k};
                        }
                    }
                    if(j+1<input.t){
                        if(dp[i+1][j+1][k]<dp[i][j][k]+score_sum[k]){
                            dp[i+1][j+1][k]=dp[i][j][k]+score_sum[k];
                            prev[i+1][j+1][k]={i,j,k};
                        }
                    }
                    if(dp[i+1][j][k]<dp[i][j][k]+50000+score_sum[k]){
                        dp[i+1][j][k]=dp[i][j][k]+50000+score_sum[k];
                        prev[i+1][j][k]={i,j,k};
                    }
                }
            }
        }
        int max_score=-INF,ni=input.t,nj=-1,nk=-1;
        rep(i,input.t+1){
            rep(j,x){
                if(max_score<dp[input.t][i][j]){
                    max_score=dp[input.t][i][j];
                    nj=i;
                    nk=j;
                }
            }
        }
        vector<int> ans;
        while(ni!=0){
            int ti,tj,tk;
            tie(ti,tj,tk)=prev[ni][nj][nk];
            if(tk!=nk){
                ans.push_back(0);
            }else if(tj!=nj){
                ans.push_back(1);
            }else{
                ans.push_back(2);
            }
            ni=ti;
            nj=tj;
            nk=tk;
        }
        reverse(all(ans));
        int now=0;
        rep(i,input.t){
            int cin_u,cin_v;
            cin >> cin_u >> cin_v;
            if(ans[i]==0){
                int s,t;
                tie(s,t)=v[now];
                int o,p,q,r;
                tie(o,p)=back(s);
                tie(q,r)=back(t);
                cout << ans[i]+1 << " " << o+1 << " " << p+1 << " " << q+1 << " " << r+1 << endl;
                now++;
            }else{
                cout << ans[i]+1 << endl;
            }
        }
    }
}solver;
int main(){
    input.input_cin();
    solver.solve();
}
0