zoukankan      html  css  js  c++  java
  • ZOJ 2866 Overstaffed Company

    树状数组

    #include<cstdio>
    #include<cstring>
    #include<cmath>
    #include<algorithm>
    #include<vector>
    using namespace std;
    
    const int maxn = 1000000 + 10;
    vector<int>Tree[50000 + 10];
    int n;
    int val[50000 + 10];
    int c[maxn];
    int ans[50000 + 10];
    
    int lowbit(int x)
    {
        return x & (-x);
    }
    
    void Update(int x, int add)
    {
        while (x < maxn)
        {
            c[x] += add;
            x += lowbit(x);
        }
    }
    
    int Get(int x)
    {
        int ret = 0;
        while (x != 0)
        {
            ret += c[x];
            x -= lowbit(x);
        }
        return ret;
    }
    
    void init()
    {
        for (int i = 0; i <= n; i++) Tree[i].clear();
        memset(c, 0, sizeof c);
    }
    
    void DFS(int now)
    {
        int A, B;
        A = Get(1000000 + 1) - Get(val[now]);
        for (int i = 0; i < Tree[now].size(); i++)
            DFS(Tree[now][i]);
        B = Get(1000000 + 1) - Get(val[now]);
        ans[now] = B - A;
        Update(val[now], 1);
    }
    
    int main()
    {
        while (~scanf("%d", &n))
        {
            init();
            for (int i = 1; i < n; i++)
            {
                int k;
                scanf("%d", &k);
                Tree[k].push_back(i);
            }
            for (int i = 0; i < n; i++)
            {
                scanf("%d", &val[i]);
                val[i]++;
            }
            DFS(0);
            for (int i = 0; i < n; i++)
            {
                if (i < n - 1) printf("%d ", ans[i]);
                else printf("%d
    ", ans[i]);
            }
        }
        return 0;
    }
  • 相关阅读:
    Python入门系列——第17篇
    Python入门系列——第16篇
    Python入门系列——第15篇
    Python入门系列——第14篇
    Python入门系列——第13篇
    Python入门系列——第12篇
    python入门系列——第11篇
    Python入门系列——第10篇
    Python入门系列——第9篇
    Python入门系列——第8篇
  • 原文地址:https://www.cnblogs.com/zufezzt/p/4779473.html
Copyright © 2011-2022 走看看