Hướng dẫn giải của Làm quen bạn mới


Chỉ dùng lời giải này khi không có ý tưởng, và đừng copy-paste code từ lời giải này. Hãy tôn trọng người ra đề và người viết lời giải.
Nộp một lời giải chính thức trước khi tự giải là một hành động có thể bị ban.

Lưu ý: Các code mẫu dưới đây chỉ mang tính tham khảo và có thể không AC được bài tập này

Code mẫu của flashmt

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <vector>
#define maxN 200200
using namespace std;

int n,k,root,maxDepth,par[maxN][20],depth[maxN];
vector <int> a[maxN],meeting[maxN/2];

void visit(int x)
{
    if (depth[x]>maxDepth) maxDepth=depth[x];
    for (int i=0;i<int(a[x].size());i++) 
    {
        depth[a[x][i]]=depth[x]+1; 
        visit(a[x][i]);
    }
}

void initLCA()
{
    for (int j=1;1<<j<=maxDepth;j++)
        for (int i=1;i<=n;i++)
            par[i][j]=par[par[i][j-1]][j-1];
}

int LCA(int x,int y)
{
    int log;
    if (depth[x]<depth[y]) swap(x,y);
    for (log=0;1<<log<=depth[x];log++);
    for (int i=log;i>=0;i--)
        if (depth[x]-(1<<i)>=depth[y]) x=par[x][i];
    if (x==y) return x;
    for (int i=log;i>=0;i--)
        if (par[x][i] && par[x][i]!=par[y][i]) x=par[x][i], y=par[y][i];
    return par[x][0];           
}

int calcMeeting(vector <int> meeting)
{
    int x=meeting[0], y, furthestDist=0, ans=0;
    for (int i=1;i<int(meeting.size());i++)
    {
        int j=meeting[i], dist=depth[x]+depth[j]-2*depth[LCA(x,j)];
        if (dist>furthestDist) furthestDist=dist, y=j;
    }
    for (int i=0;i<int(meeting.size());i++)
    {
        int j=meeting[i], dist=depth[y]+depth[j]-2*depth[LCA(y,j)];
        if (dist>ans) ans=dist;
    }
    return ans;
}

int main()
{
    int x;
    cin >> n >> k;
    for (int i=1;i<=n;i++) 
    {
        scanf("%d%d",&x,&par[i][0]);
        a[par[i][0]].push_back(i);
        meeting[x].push_back(i);
        if (!par[i][0]) root=i;
    }
    visit(root);
    initLCA();
    for (int i=1;i<=k;i++) printf("%d\n",calcMeeting(meeting[i]));
}

Code mẫu của ladpro98

#include <bits/stdc++.h>
const int N = 200205;
const int oo = 123456789;
using namespace std;
vector<int> a[N], party[N / 2];
int par[N][22], d[N], POW[22], Q[N];
int n, k, lg;

void Enter() {
    scanf("%d %d\n", &n, &k);
    int u, p, v;
    for(u=1; u<=n; u++) {
        scanf("%d %d\n", &p, &v);
        a[v].push_back(u);
        party[p].push_back(u);
    }
}

void BFS() {
    int l = 1, r = 1, i, u, v;
    Q[1] = 0;
    while (l <= r) {
        u = Q[l++];
        for(i=0; i<a[u].size(); i++) {
            v = a[u][i];
            par[v][0] = u;
            Q[++r] = v;
            d[v] = d[u] + 1;
        }
    }
}

void InitLCA() {
    lg = log2(n) + 1; int i, j;
    for(j=1; j<=lg; j++)
    for(i=1; i<=n; i++)
        par[i][j] = par[par[i][j-1]][j-1];
    POW[0] = 1;
    for(i=1; i<=lg; i++) POW[i] = POW[i-1] + POW[i-1];
}

void Jump(int& u, int x) {
    for(int i=lg; i>=0; i--)
    if (x >= POW[i]) {
        x -= POW[i];
        u = par[u][i];
    }
}

int LCA(int u, int v) {
    if (d[u] > d[v]) Jump(u, d[u] - d[v]); else Jump(v, d[v] - d[u]);
    if (u == v) return u;
    for(int i=lg; i>=0; i--)
    if (par[u][i] != par[v][i]) {
        u = par[u][i]; v = par[v][i];
    }
    return par[u][0];
}

void Process() {
    int i, j, res, deepest, v, p, u;
    for(i=1; i<=k; i++) {
        res = 0;
        deepest = party[i][0];
        for(j=1; j<party[i].size(); j++) {
            u = party[i][j];
            if (d[u] > d[deepest])
                deepest = u;
        }
        for(j=0; j<party[i].size(); j++) {
            v = party[i][j];
            p = LCA(deepest, v);
            res = max(res, d[deepest] + d[v] - d[p] - d[p]);
        }
        printf("%d\n", res);
    }
}

int main()
{
    Enter();
    BFS();
    InitLCA();
    Process();
    return 0;
}

Code mẫu của RR

#include <sstream>
#include <iomanip>
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <string>
#include <deque>
#include <complex>

#define FOR(i,a,b) for(int i=(a),_b=(b); i<=_b; i++)
#define FORD(i,a,b) for(int i=(a),_b=(b); i>=_b; i--)
#define REP(i,a) for(int i=0,_a=(a); i<_a; i++)
#define FORN(i,a,b) for(int i=(a),_b=(b);i<_b;i++)
#define DOWN(i,a,b) for(int i=a,_b=(b);i>=_b;i--)
#define SET(a,v) memset(a,v,sizeof(a))
#define sqr(x) ((x)*(x))
#define ll long long
#define F first
#define S second
#define PB push_back
#define MP make_pair

#define DEBUG(x) cout << #x << " = "; cout << x << endl;
#define PR(a,n) cout << #a << " = "; FOR(_,1,n) cout << a[_] << ' '; cout << endl;
#define PR0(a,n) cout << #a << " = "; REP(_,n) cout << a[_] << ' '; cout << endl;
using namespace std;

//Buffer reading
int INP,AM,REACHEOF;
#define BUFSIZE (1<<12)
char BUF[BUFSIZE+1], *inp=BUF;
#define GETCHAR(INP) { \
    if(!*inp) { \
        if (REACHEOF) return 0;\
        memset(BUF,0,sizeof BUF);\
        int inpzzz = fread(BUF,1,BUFSIZE,stdin);\
        if (inpzzz != BUFSIZE) REACHEOF = true;\
        inp=BUF; \
    } \
    INP=*inp++; \
}
#define DIG(a) (((a)>='0')&&((a)<='9'))
#define GN(j) { \
    AM=0;\
    GETCHAR(INP); while(!DIG(INP) && INP!='-') GETCHAR(INP);\
    if (INP=='-') {AM=1;GETCHAR(INP);} \
    j=INP-'0'; GETCHAR(INP); \
    while(DIG(INP)){j=10*j+(INP-'0');GETCHAR(INP);} \
    if (AM) j=-j;\
}
//End of buffer reading

const long double PI = acos((long double) -1.0);

const int MN = 200111;

int father[22][MN], h[MN];
vector<int> ke[MN], ls[MN];
int n, k;

void dfs(int u) {
    REP(i,ke[u].size()) {
        int v = ke[u][i];
        father[0][v] = u;
        h[v] = h[u] + 1;
        dfs(v);
    }
}

void init() {
    FOR(i,1,18)
    FOR(u,1,n) {
        int v = father[i-1][u];
        if (v) father[i][u] = father[i-1][v];
    }
}

#define TWO(x) (1<<(x))

int get(int u, int v) {
    if (h[u] < h[v]) swap(u, v);
    int res = 0;
    if (h[u] > h[v]) {
        FORD(i,18,0) {
            int x = father[i][u];
            if (x && h[x] >= h[v]) {
                res += TWO(i);
                u = x;
            }
        }
    }
    if (u != v) {
        FORD(i,18,0)
        if (father[i][u] != father[i][v]) {
            res += TWO(i) + TWO(i);
            u = father[i][u];
            v = father[i][v];
        }
        res += 2;
    }
    return res;
}

int main() {
    scanf("%d%d", &n, &k);
    int root = 1;
    FOR(i,1,n) {
        int c, u;
        scanf("%d%d", &c, &u);
        ls[c].PB(i);
        if (u) ke[u].PB(i);
        else root = i;
    }
    dfs(root);
    init();
    // PR(h, n);

    FOR(c,1,k) {
        int x = ls[c][0], y, best = 0, save = x, cur;

        REP(i,ls[c].size()) {
            y = ls[c][i];
            cur = get(x, y);

            if (cur > best) {
                best = cur;
                save = y;
            }
        }

        x = save, best = 0, save = x;

        REP(i,ls[c].size()) {
            y = ls[c][i];
            cur = get(x, y);

            if (cur > best) {
                best = cur;
                save = y;
            }
        }

        printf("%d\n", best);
    }
    return 0;
}

Bình luận

Hãy đọc nội quy trước khi bình luận.



  • 0
    21ti_hdhphat  đã bình luận lúc 31, Tháng 5, 2022, 2:10 chỉnh sửa

    Code mẫu của "flashmt" và "RR": Mình chưa hiểu ý tưởng khúc tính calcMeeting() cho lắm. Tại sao nó lại đúng? Theo mình hiểu là lúc đầu, tính nút xa nhất từ meeting[0], gọi nút đó là "y". Sau đó tính nút xa nhất từ nút "y". Khoảng cách giữa "y" tới nút xa nhất chính là câu trả lời. Hmm... Tại sao nó lại đúng vậy nhỉ? Giải thích giúp mình với, cảm ơn nhiều ạ!


    • 1
      hh123123  đã bình luận lúc 31, Tháng 5, 2022, 13:43

      bạn đọc sol tại đây nhé

      https://vnoi.info/wiki/algo/data-structures/lca-binlift.md#b%C3%A0i-to%C3%A1n-2-1