poj 1470 | zoj 1141 Closest Common Ancestors

32 篇文章 0 订阅
13 篇文章 0 订阅

类型:LCA

题目:http://poj.org/problem?id=1470

来源:Southeastern Europe 2000

思路:使用LCA算法求最近公共祖先,并记录个数

// poj 1470 Closest Common Ancestors
// OLE AC 2956K	563MS
#include <iostream>
#include <string>
#include <queue>
#include <stack>
#include <vector>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
using namespace std;

#define FOR(i,a,b) for(i = (a); i < (b); ++i)
#define FORE(i,a,b) for(i = (a); i <= (b); ++i)
#define FORD(i,a,b) for(i = (a); i > (b); --i)
#define FORDE(i,a,b) for(i = (a); i >= (b); --i)
#define CLR(a,b) memset(a,b,sizeof(a))
#define PB(x) push_back(x)

const int MAXN = 1100;
const int MAXM = 210000;

bool vis[MAXN];
int cnt, cntt, n;
int head[MAXN], p[MAXN], ru[MAXN], ancestor[MAXN], endd[MAXN];
vector<int> q[MAXN];
struct edge {
    int v, nxt;
}e[MAXM];

void addedge(int u, int v) {
    e[cnt].v = v;
    e[cnt].nxt = head[u];
    head[u] = cnt++;
}

void makeSet() {
	for(int i = 1; i <= n; ++i)
		p[i] = i;
}

int find(int u) {
	return (u != p[u]) ? (p[u] = find(p[u])) : p[u];
}

void link(int x, int y) {
    p[x] = y;
}

void LCA(int u) {
    int v, i;

    ancestor[u] = u;
    for(i = head[u]; i != -1; i = e[i].nxt) {
        v = e[i].v;
        LCA(v);
        link(find(v), find(u));
        ancestor[find(u)] = u;
    }
    vis[u] = true;
    int size = q[u].size();
    FOR(i, 0, size)
        if(vis[q[u][i]])
            ++endd[ancestor[find(q[u][i])]];
}

void init() {
    int i;

    CLR(head, -1);
    CLR(p, 0);
    CLR(endd, 0);
    CLR(ru, 0);
    CLR(ancestor, 0);
    CLR(vis, false);
    cnt = cntt = 0;
    //!!!
    makeSet();
    FORE(i, 1, n)
        q[i].clear();
}

int main() {
    int i, j, u, v, num;

    while(scanf("%d", &n) != EOF) {
        init();
        FORE(i, 1, n) {
            scanf("%d:(%d)", &u, &num);
            FOR(j, 0, num) {
                scanf("%d", &v);
                addedge(u, v);
                ++ru[v];
            }
        }
        scanf("%d", &num);
        FORE(j, 1, num) {
            scanf(" (%d %d)", &u, &v);
            q[u].push_back(v);
            q[v].push_back(u);
        }
        FORE(i, 1, n) if(ru[i] == 0) {
            LCA(i);
            break;
        }
        FORE(i, 1, n) if(endd[i] != 0)
            printf("%d:%d\n", i, endd[i]);
    }
    return 0;
}




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值