EOJ Mouth 2018.5 C. 面向对象编程 - 思维

Time limit per test: 3.0 seconds

Memory limit: 1024 megabytes

在面向对象程序设计中，常常会运用到函数的扩展与重写。当一个类继承某个类的时候，它可以调用所有父类可以调用的函数。它可以声明新的函数。当新的函数签名与父类的某个函数一致时，就会发生函数的覆盖（重写）。所以，在子类的实例调用某个函数时，它会调用最近的父类（有可能是它自己）的那个函数实现。

这里我们不考虑访问权限等情况，我们只关心某个类在调用某个函数时，这个函数是在哪个类中实现的。

Input

输入具有如下格式：

np2 p3 … pnt1 a11 a12 … a1t1t2 a21 a22 … a2t2⋮tn an1 an2 … antnqu1 r1u2 r2⋮uq rq

解释与数据规模约定：

n 表示有 n 个类，这些类从 1 到 n 编号。2≤n≤105。
pi 表示第 i 个类的父类编号。1≤pi≤i−1。第 1 个类是所有类共同的祖先类，这个类没有父类。
ti 表示在第 i 个类中定义了多少个函数，ai1,ai2,…,aiti 表示第 i 个类中的函数列表。同一个类的函数列表中不会出现两个相同的函数。1≤aij≤106。0≤ti≤106,∑ti≤106。
q 表示询问个数。1≤q≤105。
ui,ri 表示第 i 个询问，询问第 ui 个类的实例在调用 ri 函数时，调用的是哪个类中的版本。1≤ui≤n,1≤ri≤106。

Output

对于每个询问，输出答案。如果调用不合法（会导致编译错误），输出 −1。

Examples

Note

样例等价于下面的 Java 代码。

class Class1 {
    void function2() { System.out.println("1"); }
    void function1() { System.out.println("1"); }
}

class Class2 extends Class1 {

}

class Class3 extends Class2 {
    void function5() { System.out.println("3"); }
    void function2() { System.out.println("3"); }
}

class Class4 extends Class3 {
    void function4() { System.out.println("4"); }
    void function5() { System.out.println("4"); }
}

class Class5 extends Class3 {
    void function5() { System.out.println("5"); }
}

void test() {
    new Class3().function4();
    new Class5().function2();
    new Class4().function5();
    new Class1().function3();
}

题解：

离线处理. 对每种函数维护一个栈, 栈中记录类名. DFS 的时候更新栈 (进入时压入, 退出时弹出). 对于某个节点上的所有查询, 答案就是对应函数的栈顶元素.

给了一个有n个节点的树，保证根节点为1,每一个节点可能控制着若干函数，最后有q组询问，每组询问查询当前节点a的第b个函数被哪个节点控制，查询不合法就输出-1.

我们用邻接表先把这个树和每一个查询存起来，然后用一遍dfs，在遍历整个树的时候先查看当前节点控制着哪些函数，然后添加在对应的栈中去，然后查看当前节点是否有询问，如果有的时候就去栈中找答案。然后继续dfs，在回溯的时候进行出栈操作吗，整个过程是离线的，只需要对一棵树进行一遍dfs即可

以上操作就模拟了函数的调用过程

代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5+10;
struct node {
    int funs,id;
};
vector<struct node> query[maxn];
vector<int> Store[maxn],way[maxn];
stack<int> Fun[maxn*10];
int arr[maxn],ans[maxn];

void dfs(int rt)
{
    for(auto v:Store[rt])
        Fun[v].push(rt);

    for(auto v:query[rt]) {
        if(Fun[v.funs].empty()) ans[v.id] = -1;
        else ans[v.id] = Fun[v.funs].top();
    }       
    for(auto v:way[rt]) dfs(v);
    for(auto v:Store[rt]) Fun[v].pop();
}
inline void init() 
{
    for(int i=0;i<maxn;i++) {
        way[i].clear();Store[i].clear();query[i].clear();
        while(!Fun[i].empty()) Fun[i].pop();
    }
    
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        init();
        for(int i=2,x;i<=n;i++) {
            scanf("%d",&x);
            way[x].push_back(i);
        }

        for(int i=1,x,y;i<=n;i++) {
            scanf("%d",&x);
            while(x--) {
                scanf("%d",&y);
                Store[i].push_back(y);
            }
        }

        int q,u;
        node temp ;
        scanf("%d",&q);
        for(int i=1;i<=q;i++) {
            scanf("%d%d",&u,&temp.funs);
            temp.id = i;
            query[u].push_back(temp);
        }

        dfs(1);

        for(int i=1;i<=q;i++) printf("%d\n",ans[i]); 
    }
    return 0;
}