zoukankan      html  css  js  c++  java
  • Bzoj3631: [JLOI2014]松鼠的新家

    树剖后在dfn上差分

    # include <bits/stdc++.h>
    # define RG register
    # define IL inline
    # define Fill(a, b) memset(a, b, sizeof(a))
    using namespace std;
    typedef long long ll;
    const int _(3e5 + 10);
    
    IL ll Read(){
        RG ll x = 0, z = 1; RG char c = getchar();
        for(; c < '0' || c > '9'; c = getchar()) z = c == '-' ? -1 : 1;
        for(; c >= '0' && c <= '9'; c = getchar()) x = (x << 1) + (x << 3) + (c ^ 48);
        return x * z;
    }
    
    int n, fst[_], nxt[_ << 1], to[_ << 1], cnt, a[_], ans[_];
    int dfn[_], Index, fa[_], size[_], top[_], son[_], deep[_];
    
    IL void Add(RG int u, RG int v){  to[cnt] = v; nxt[cnt] = fst[u]; fst[u] = cnt++;  }
    
    IL void Dfs1(RG int u, RG int ff){
        size[u] = 1; fa[u] = ff; deep[u] = deep[ff] + 1;
        for(RG int e = fst[u]; e != -1; e = nxt[e]){
            if(to[e] == ff) continue;
            Dfs1(to[e], u);
            size[u] += size[to[e]];
            if(size[to[e]] > size[son[u]]) son[u] = to[e];
        }
    }
    
    IL void Dfs2(RG int u, RG int Top){
        top[u] = Top; dfn[u] = ++Index;
        if(son[u]) Dfs2(son[u], Top);
        for(RG int e = fst[u]; e != -1; e = nxt[e]) if(!dfn[to[e]]) Dfs2(to[e], to[e]);
    }
    
    IL void Cover(RG int u, RG int v){
        while(top[u] ^ top[v]){
            if(deep[top[u]] < deep[top[v]]) swap(u, v);
    		++ans[dfn[top[u]]]; --ans[dfn[u] + 1];
            u = fa[top[u]];
        }
        if(deep[u] > deep[v]) swap(u, v);
    	++ans[dfn[u]]; --ans[dfn[v] + 1];
    }
    
    int main(RG int argc, RG char* argv[]){
        n = Read();
    	for(RG int i = 1; i <= n; ++i) a[i] = Read(), fst[i] = -1;
    	for(RG int i = 1, x, y; i < n; ++i) x = Read(), y = Read(), Add(x, y), Add(y, x);
    	Dfs1(1, 0); Dfs2(1, 1);
    	for(RG int i = 1; i < n; ++i) Cover(a[i], a[i + 1]);
    	for(RG int i = 1; i <= n; ++i) ans[i] += ans[i - 1];
    	for(RG int i = 1; i <= n; ++i) printf("%d
    ", ans[dfn[i]] - (i != a[1]));
        return 0;
    }
    
    
  • 相关阅读:
    Selenium-Xpath使用方法
    HTML基础之js
    HTML基础—DOM操作
    CSS基础知识总结二
    CSS基础知识总结之css样式引用的三种方式
    BZOJ 2329/2209 [HNOI2011]括号修复 (splay)
    BZOJ 1576 [USACO]安全路经Travel (树剖+线段树)
    BZOJ 2402 陶陶的难题II (01分数规划+树剖+线段树+凸包+二分)
    BZOJ 4044 Virus synthesis (回文自动机+dp)
    BZOJ 2342 [SHOI2011]双倍回文 (回文自动机)
  • 原文地址:https://www.cnblogs.com/cjoieryl/p/8289985.html
Copyright © 2011-2022 走看看