結果

問題 No.1778 括弧列クエリ / Bracketed Sequence Query
コンテスト
ユーザー houren
提出日時 2021-12-08 09:40:34
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 573 ms / 2,000 ms
コード長 2,180 bytes
コンパイル時間 1,867 ms
コンパイル使用メモリ 188,712 KB
実行使用メモリ 26,652 KB
最終ジャッジ日時 2024-07-16 01:53:05
合計ジャッジ時間 15,253 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using P = pair<int,int>;
#define rep(i, n) for(int i = 0; i < n; i++)
#define all(x) (x).begin(),(x).end()
struct Edge{ ll to; };
using G = vector<vector<Edge>>;
struct LCA{
    vector<vector<int>> par;
    vector<int> dist;
    LCA(const G &g, int root = 0) { init(g, root); }

    void init(const G g, int root){
        int n = g.size(), k = 1;
        while((1 << k) < n) k++;
        par.assign(k,vector<int>(n,-1));
        dist.assign(n,-1);
        dfs(g, root, -1, 0);
        rep(i,k-1){
            rep(j,n){
                if(par[i][j] < 0) par[i+1][j] = -1;
                else par[i+1][j] = par[i][par[i][j]];
            }
        }
    }
    void dfs(const G &g, int now, int p, int d){
        par[0][now] = p;
        dist[now] = d;
        for(auto x : g[now]){
            if(x.to != p) dfs(g, x.to, now, d+1);
        }
    }
    int query(int u, int v){
        if(dist[u] < dist[v]) swap(u, v);
        int k = par.size();
        rep(i,k){
            if((dist[u]-dist[v]) >> i & 1) u = par[i][u];
        }
        if(u==v) return u;
        for(int i=k-1;i>=0;i--){
            if(par[i][u] != par[i][v]){
                u = par[i][u];
                v = par[i][v];
            }
        }
        return par[0][u];
    }
};
int main(){
    int n,q;
    string s;
    cin >> n >> q >> s;
    stack<P> sta;
    sta.push({0,-1});
    G g(n/2+1);
    vector<int> num(n);
    vector<P> numinv(n/2+1);
    int now = 0;
    rep(i,n){
        if(s[i] == '('){
            now++;
            num[i] = now;
            Edge e1, e2;
            e1.to = sta.top().first; e2.to = now;
            g[e1.to].push_back(e2);
            g[e2.to].push_back(e1);
            sta.push({now,i});
        }else{
            num[i] = sta.top().first;
            numinv[sta.top().first] = {sta.top().second, i};
            sta.pop();
        }
    }
    LCA lca(g,0);
    rep(i,q){
        int u,v;
        cin >> u >> v;
        auto x = lca.query(num[u-1],num[v-1]);
        if(!x) cout << -1 << endl;
        else cout << numinv[x].first+1 << " " << numinv[x].second+1 << endl;
    }
    return 0;
}
0