UVALIVE 3486 Cells

最新推荐文章于 2018-10-16 17:03:01 发布

deko2014

最新推荐文章于 2018-10-16 17:03:01 发布

阅读量132

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Commence/p/4048012.html

本文介绍了一种使用栈和深度优先搜索(DFS)算法来判断二叉树中节点间祖先关系的方法。通过预处理节点的入栈和出栈顺序，可以高效地判断任意两个节点之间的祖先关系。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

通过入栈出栈顺序判断祖先关系

这里UVALIVE还

#include <map>
#include <set>
#include <list>
#include <cmath>
#include <ctime>
#include <deque>
#include <stack>
#include <queue>
#include <cctype>
#include <cstdio>
#include <string>
#include <vector>
#include <climits>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define LL long long
#define PI 3.1415926535897932626
using namespace std;
int gcd(int a, int b) {return a % b == 0 ? b : gcd(b, a % b);}
#define MAXN 20000010
int pre[MAXN],post[MAXN];
int start[300005],c[300005];
int N;
void dfs(int root)
{
        stack<int>s;
        s.push(root);
        pre[root] = 0;
        int dfs_clock = 0;
        while (!s.empty())
        {
                int x = s.top();
                if (pre[x])
                {
                        post[x] = ++dfs_clock;
                        s.pop();
                        continue;
                }
                pre[x] = ++dfs_clock;
                for (int i =start[x]; i < start[x] + c[x] ; i++)
                {
                        if (i < N)
                        {
                                pre[i] = 0 ;
                                s.push(i);
                        }
                        else
                        {
                                pre[i] = ++dfs_clock;
                                post[i] = ++dfs_clock;
                        }
                }
        }
}
int main()
{
        int T ,kase = 1;
        scanf("%d",&T);
        while (T--)
        {
                printf("Case %d:\n",kase++);
                scanf("%d",&N);
                start[0] = 1;
                for (int i = 0; i < N; i++)
                {
                        scanf("%d",&c[i]);
                        if (i)
                                start[i] = start[i - 1] + c[i - 1];
                }
                dfs(0);
                int m;
                scanf("%d",&m);
                while (m--)
                {
                        int u,v;
                        scanf("%d%d",&u,&v);
                        if(pre[u] < pre[v] && post[u] > post[v]) puts("Yes");
                        else puts("No");
                }
                if (T) putchar('\n');
        }
}