2016湖南省赛 I Tree Intersection（线段树合并，树链剖分）
2016湖南省赛 I Tree Intersection（线段树合并，树链剖分）
传送门：https://ac.nowcoder.com/acm/contest/1112/I
题意：
给你一个n个结点的树，树上每个节点有自己的颜色
问你删除第i条边后形成的两颗子树有多少个相同的颜色
题解：
树链剖分写法：
对于每一种颜色来说，如果这个颜色是在单独的一颗子树中，那么就不会对其他的边产生贡献，所以我们单独对每一种颜色对边的贡献讨论，如果这个颜色只有一个，那么就不会产生贡献，否则，他就可以在两个相同颜色之间的路径上的边产生贡献，用树剖剖下来后区间加一加就可以了
线段树合并的写法
对于每一种颜色建权值线段树，线段树合并统计答案即可
代码：
树链剖分写法：
/**
* ┏┓ ┏┓
* ┏┛┗━━━━━━━┛┗━━━┓
* ┃ ┃
* ┃ ━ ┃
* ┃ ＞ ＜ ┃
* ┃ ┃
* ┃... ⌒ ... ┃
* ┃ ┃
* ┗━┓ ┏━┛
* ┃ ┃ Code is far away from bug with the animal protecting
* ┃ ┃ 神兽保佑,代码无bug
* ┃ ┃
* ┃ ┃
* ┃ ┃
* ┃ ┃
* ┃ ┗━━━┓
* ┃ ┣┓
* ┃ ┏┛
* ┗┓┓┏━┳┓┏┛
* ┃┫┫ ┃┫┫
* ┗┻┛ ┗┻┛
*/
// warm heart, wagging tail,and a smile just for you!
//
// _ooOoo_
// o8888888o
// 88" . "88
// (| -_- |)
// O\ = /O
// ____/`---'\____
// .' \| |// `.
// / \||| : |||// \
// / _||||| -:- |||||- \
// | | \ - /// | |
// | \_| ''\---/'' | |
// \ .-\__ `-` ___/-. /
// ___`. .' /--.--\ `. . __
// ."" '< `.___\_<|>_/___.' >'"".
// | | : `- \`.;`\ _ /`;.`/ - ` : | |
// \ \ `-. \_ __\ /__ _/ .-` / /
// ======`-.____`-.___\_____/___.-`____.-'======
// `=---='
// ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
// 佛祖保佑 永无BUG
#include <set>
#include <map>
#include <stack>
#include <cmath>
#include <queue>
#include <cstdio>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long LL;
typedef pair<int, int> pii;
typedef unsigned long long uLL;
#define ls rt<<1
#define rs rt<<1|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define bug printf("*********\n")
#define FIN freopen("input.txt","r",stdin);
#define FON freopen("output.txt","w+",stdout);
#define IO ios::sync_with_stdio(false),cin.tie(0)
#define debug1(x) cout<<"["<<#x<<" "<<(x)<<"]\n"
#define debug2(x,y) cout<<"["<<#x<<" "<<(x)<<" "<<#y<<" "<<(y)<<"]\n"
#define debug3(x,y,z) cout<<"["<<#x<<" "<<(x)<<" "<<#y<<" "<<(y)<<" "<<#z<<" "<<z<<"]\n"
const int maxn = 3e5 + 5;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;
const double Pi = acos(-1);
LL gcd(LL a, LL b) {
return b ? gcd(b, a % b) : a;
}
LL lcm(LL a, LL b) {
return a / gcd(a, b) * b;
}
double dpow(double a, LL b) {
double ans = 1.0;
while(b) {
if(b % 2)ans = ans * a;
a = a * a;
b /= 2;
} return ans;
}
LL quick_pow(LL x, LL y) {
LL ans = 1;
while(y) {
if(y & 1) {
ans = ans * x % mod;
} x = x * x % mod;
y >>= 1;
} return ans;
}
int n;
int a[maxn], sz[maxn], dep[maxn], fa[maxn], top[maxn], w[maxn], son[maxn], W[maxn];
int sum[maxn << 2], lazy[maxn << 2], vis[maxn << 2];
struct EDGE {
int u, v, nt;
} edge[maxn << 1];
int head[maxn], summ, cnt;
int u[maxn];
int v[maxn];
void add_edge(int u, int v) {
edge[++summ].u = u;
edge[summ].v = v;
edge[summ].nt = head[u];
head[u] = summ;
}
void dfs1(int u) {
sz[u] = 1; son[u] = 0;
for (int i = head[u]; ~i; i = edge[i].nt) {
int v = edge[i].v;
if (v != fa[u]) {
fa[v] = u;
dep[v] = dep[u] + 1;
dfs1(v);
sz[u] += sz[v];
if (sz[v] > sz[son[u]]) son[u] = v;
}
}
}
void dfs2(int u, int tp, int x) {
top[u] = tp; w[u] = ++cnt; W[cnt] = u;
if (son[u]) dfs2(son[u], tp, 1);
for (int i = head[u]; ~i; i = edge[i].nt) {
int v = edge[i].v;
if (v == son[u] || v == fa[u]) continue;
dfs2(v, v, 2);
}
}
vector<int> vec[maxn];
void init() {
memset(head, -1, sizeof(head));
summ = 1; cnt = 0;
for(int i = 1; i <= n; i++) {
vec[i].clear();
}
for (int i = 1; i <= n; i++) {
scanf("%d", &a[i]);
vec[a[i]].push_back(i);
}
for (int i = 1; i < n; i++) {
// int u, v;
scanf("%d %d", &u[i], &v[i]);
add_edge(u[i], v[i]);
add_edge(v[i], u[i]);
}
dep[1] = 1; fa[1] = 0;
dfs1(1);
dfs2(1, 1, 1);
}
void pushup(int rt) {
if(vis[ls] == vis[rs]) vis[rt] = vis[ls];
else vis[rt] = -1;
}
void pushdown(int rt, int mid) {
if (lazy[rt]) {
lazy[rt << 1] = lazy[rt];
lazy[rt << 1 | 1] = lazy[rt];
vis[rt << 1] = 0;
vis[rt << 1 | 1] = 0;
lazy[rt] = 0;
}
}
void build(int l, int r, int rt) {
lazy[rt] = 0;
sum[rt] = 0;
if (l == r) {
vis[rt] = 0;
return;
}
int mid = (l + r) >> 1;
build(lson);
build(rson);
}
void update(int L, int R, int val, int l, int r, int rt) {
// debug2(L, R);
if(L > R || r < L || R < l) return;
if(vis[rt] == 1) return;
if (L <= l && r <= R) {
// sum[rt] += val * (r - l + 1);
// lazy[rt] += val;
if(vis[rt] == 0) {
vis[rt] = 1;
sum[rt]++;
return;
}
}
pushdown(rt, r - l + 1);
int mid = (l + r) >> 1;
if (L <= mid) update(L, R, val, lson);
if (R > mid) update(L, R, val, rson);
pushup(rt);
}
int query(int pos, int l, int r, int rt) {
if (l == r) {
return sum[rt];
}
pushdown(rt, r - l + 1);
int mid = (l + r) >> 1;
int ans = sum[rt];
if (pos <= mid) ans += query(pos, lson);
else ans += query(pos, rson);
return ans;
}
int LCA(int x, int y) {
// debug2(x, y);
while (top[x] != top[y]) {
// bug;
if (dep[top[x]] < dep[top[y]]) swap(x, y);
update(w[top[x]], w[x], 1, 1, n, 1);
x = fa[top[x]];
}
if (dep[x] > dep[y])
std::swap(x, y);
update(w[x] + 1, w[y], 1, 1, n, 1);
return x;
}
int main() {
#ifndef ONLINE_JUDGE
FIN
#endif
while(scanf("%d", &n) != EOF) {
init();
build(1, n, 1);
// for(int i = 1; i <= n; i++) {
// // printf("%d\n", w[]);
// // debug1(w[i]);
// }
for(int i = 1; i <= n; i++) {
vis[1] = 0;
lazy[1] = 1;
int sz = vec[i].size();
if(sz < 2) continue;
int bb = LCA(vec[i][0], vec[i][1]);
for(int j = 2; j < sz; j++) {
// debug2(vec[i][0], vec[i][1]);
bb = LCA(bb, vec[i][j]);
}
}
for(int i = 1; i < n; i++) {
int ans;
// debug2(v[i], u[i]);
if(fa[v[i]] == u[i]) {
ans = query(w[v[i]], 1, n, 1);
} else {
ans = query(w[u[i]], 1, n, 1);
}
printf("%d\n", ans);
}
}
return 0;
}
线段树合并写法
/**
* ┏┓ ┏┓
* ┏┛┗━━━━━━━┛┗━━━┓
* ┃ ┃
* ┃ ━ ┃
* ┃ ＞ ＜ ┃
* ┃ ┃
* ┃... ⌒ ... ┃
* ┃ ┃
* ┗━┓ ┏━┛
* ┃ ┃ Code is far away from bug with the animal protecting
* ┃ ┃ 神兽保佑,代码无bug
* ┃ ┃
* ┃ ┃
* ┃ ┃
* ┃ ┃
* ┃ ┗━━━┓
* ┃ ┣┓
* ┃ ┏┛
* ┗┓┓┏━┳┓┏┛
* ┃┫┫ ┃┫┫
* ┗┻┛ ┗┻┛
*/
// warm heart, wagging tail,and a smile just for you!
//
// _ooOoo_
// o8888888o
// 88" . "88
// (| -_- |)
// O\ = /O
// ____/`---'\____
// .' \| |// `.
// / \||| : |||// \
// / _||||| -:- |||||- \
// | | \ - /// | |
// | \_| ''\---/'' | |
// \ .-\__ `-` ___/-. /
// ___`. .' /--.--\ `. . __
// ."" '< `.___\_<|>_/___.' >'"".
// | | : `- \`.;`\ _ /`;.`/ - ` : | |
// \ \ `-. \_ __\ /__ _/ .-` / /
// ======`-.____`-.___\_____/___.-`____.-'======
// `=---='
// ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
// 佛祖保佑 永无BUG
#include <set>
#include <map>
#include <stack>
#include <cmath>
#include <queue>
#include <cstdio>
#include <string>
#include <bitset>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long LL;
typedef pair<int, int> pii;
typedef unsigned long long uLL;
#define ls rt<<1
#define rs rt<<1|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define bug printf("*********\n")
#define FIN freopen("input.txt","r",stdin);
#define FON freopen("output.txt","w+",stdout);
#define IO ios::sync_with_stdio(false),cin.tie(0)
#define debug1(x) cout<<"["<<#x<<" "<<(x)<<"]\n"
#define debug2(x,y) cout<<"["<<#x<<" "<<(x)<<" "<<#y<<" "<<(y)<<"]\n"
#define debug3(x,y,z) cout<<"["<<#x<<" "<<(x)<<" "<<#y<<" "<<(y)<<" "<<#z<<" "<<z<<"]\n"
const int maxn = 1e5 + 5;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;
const double Pi = acos(-1);
LL gcd(LL a, LL b) {
return b ? gcd(b, a % b) : a;
}
LL lcm(LL a, LL b) {
return a / gcd(a, b) * b;
}
double dpow(double a, LL b) {
double ans = 1.0;
while(b) {
if(b % 2)ans = ans * a;
a = a * a;
b /= 2;
} return ans;
}
LL quick_pow(LL x, LL y) {
LL ans = 1;
while(y) {
if(y & 1) {
ans = ans * x % mod;
} x = x * x % mod;
y >>= 1;
} return ans;
}
int col[maxn];
int cnt[maxn];
struct EDGE {
int v, nxt;
} edge[maxn << 1];
int head[maxn], tot;
void init() {
memset(head, -1, sizeof(head));
tot = 0;
}
void add_edge(int u, int v) {
edge[tot].v = v;
edge[tot].nxt = head[u];
head[u] = tot++;
}
struct node {
int l, r, sum, val;
} tree[maxn * 20];
int root[maxn];
int rt_cnt = 0;
int n;
int ans[maxn];
void push_up(int rt) {
tree[rt].sum = tree[tree[rt].l].sum + tree[tree[rt].r].sum;
}
int build(int l, int r, int pos) {
int rt = ++rt_cnt;
tree[rt].l = 0;
tree[rt].r = 0;
tree[rt].sum = 0;
if(l == r) {
tree[rt].val = 1;
if(tree[rt].val != cnt[l]) {
tree[rt].sum = 1;
} else {
tree[rt].sum = 0;
}
return rt;
}
int mid = (l + r) >> 1;
if(pos <= mid) tree[rt].l = build(l, mid, pos);
else tree[rt].r = build(mid + 1, r, pos);
push_up(rt);
return rt;
}
void merge(int &x, int y, int l, int r) {
if(!x || !y) {
if(!x) x = y;
return;
}
if(l == r) {
tree[x].val += tree[y].val;
tree[x].sum = (tree[x].val != cnt[l]);
return;
}
int mid = (l + r) >> 1;
merge(tree[x].l, tree[y].l, l, mid);
merge(tree[x].r, tree[y].r, mid + 1, r);
push_up(x);
}
void dfs(int u, int fa, int id) {
root[u] = build(1, n, col[u]);
for(int i = head[u]; i != -1; i = edge[i].nxt) {
int v = edge[i].v;
if(v == fa) continue;
dfs(v, u, i);
merge(root[u], root[v], 1, n);
}
if(u != 1) {
int tid = id / 2 + 1;
ans[tid] = tree[root[u]].sum;
}
}
int main() {
#ifndef ONLINE_JUDGE
FIN
#endif
scanf("%d", &n);
for(int i = 1; i <= n; i++) {
scanf("%d", &col[i]);
// vec[col[i]].push_back(i);
cnt[col[i]]++;
}
init();
for(int i = 1, u, v; i < n; i++) {
scanf("%d%d", &u, &v);
add_edge(u, v);
add_edge(v, u);
}
// debug1(n);
dfs(1, -1, -1);
for(int i = 1; i <= n - 1; i++) {
printf("%d\n", ans[i]);
}
return 0;
}
2016湖南省赛 I Tree Intersection（线段树合并，树链剖分）的更多相关文章
- 2016湖南省赛----G - Parenthesis （括号匹配）
2016湖南省赛----G - Parenthesis (括号匹配) Bobo has a balanced parenthesis sequence P=p 1 p 2…p n of lengt ...
- 2016湖南省赛----A 2016 （同余定理）
2016湖南省赛----A 2016 (同余定理) Description 给出正整数 n 和 m,统计满足以下条件的正整数对 (a,b) 的数量: 1. 1≤a≤n,1≤b≤m; 2. a×b 是 ...
- 2016湖南省赛 [Cloned]
A.2016 给出正整数 n 和 m,统计满足以下条件的正整数对 (a,b) 的数量: 1. 1≤a≤n,1≤b≤m; 2. a×b 是 2016 的倍数. Input 输入包含不超过 30 组数 ...
- 线段树合并 || 树状数组 || 离散化 || BZOJ 4756: [Usaco2017 Jan]Promotion Counting || Luogu P3605 [USACO17JAN]Promotion Counting晋升者计数
题面:P3605 [USACO17JAN]Promotion Counting晋升者计数 题解:这是一道万能题,树状数组 || 主席树 || 线段树合并 || 莫队套分块 || 线段树 都可以写..记 ...
- 【洛谷4482】Border的四种求法（后缀自动机_线段树合并_链分治）
这题我写了一天后交了一发就过了我好兴奋啊啊啊啊啊啊 题目 洛谷 4482 分析 这题明明可以在线做的,为什么我见到的所有题解都是离线啊 -- 什么时候有机会出一个在线版本坑人. 题目的要求可以转化为求 ...
- BZOJ.3653.谈笑风生(长链剖分/线段树合并/树状数组)
BZOJ 洛谷 \(Description\) 给定一棵树,每次询问给定\(p,k\),求满足\(p,a\)都是\(b\)的祖先,且\(p,a\)距离不超过\(k\)的三元组\(p,a,b\)个数. ...
- bzoj5518 & loj3046 「ZJOI2019」语言 线段树合并+树链的并
题目传送门 https://loj.ac/problem/3046 题解 首先问题就是问有多少条路径是给定的几条路径中的一条的一个子段. 先考虑链的做法. 枚举右端点 \(i\),那么求出 \(j\) ...
- luogu P6088 [JSOI2015]字符串树 可持久化trie 线段树合并 树链剖分 trie树
LINK:字符串树 先说比较简单的正解.由于我没有从最简单的考虑答案的角度思考 所以... 下次还需要把所有角度都考察到. 求x~y的答案 考虑 求x~根+y~根-2*lca~根的答案. 那么问题变成 ...
- [FJOI2014]最短路径树问题 长链剖分
[FJOI2014]最短路径树问题 LG传送门 B站传送门 长链剖分练手好题. 如果你还不会长链剖分的基本操作,可以看看我的总结. 这题本来出的很没水平,就是dijkstra(反正我是不用SPFA)的 ...
随机推荐
- Data Lake Analytics中OSS LOCATION的使用说明
前言 Data Lake Analytic(后文简称 DLA)可以帮助用户通过标准的SQL语句直接对存储在OSS.TableStore上的数据进行查询分析. 在查询前,用户需要根据数据文件的格式和内容 ...
- Linux常用命令1 文件处理命令
1.命令格式 1.用中括号括起来的内容都不是必填内容,碧如上图的选项和参数,有些命令不写选项和参数也可以执行 2.注意图中的简化选项与完整选项说明,完整选项要两个横杆-- 2.目录处理命令ls 1.文 ...
- Otracle数据库定时任务--dbms_job
一.dbms_job涉及到的知识点 1.创建job: variable jobno number; dbms_job.submit(:jobno, --job号 'your_procedure;'-- ...
- 小爬爬4:selenium操作
1.selenium是什么? selenium: - 概念:是一个基于浏览器自动化的模块. - 和爬虫之间的关联? - 帮我我们便捷的爬取到页面中动态加载出来的数据 - 实现模拟登陆 - 基本使用流程 ...
- js数组增删
1.shift() 2.pop() 3.push() 4.unshift() 5.splice(start,num,string...)
- spoj SUBLEX (Lexicographical Substring Search) RE的欢迎来看看
SPOJ.com - Problem SUBLEX 这么裸的一个SAM,放在了死破OJ上面就是个坑. 注意用SAM做的时候输出要用一个数组存下来,然后再puts,不然一个一个字符输出会更慢. 还有一个 ...
- 微信小程序 mode 的几种模式
mode="aspectFill" mode 有效值: mode 有 13 种模式,其中 4 种是缩放模式,9 种是裁剪模式. 模式 值 说明缩放 scaleToFill 不保持纵 ...
- 屏蔽指定地区IP访问
<?php if ($HTTP_SERVER_VARS["HTTP_X_FORWARDED_FOR"]) { $ip = $HTTP_SERVER_VARS["HT ...
- Java反射机制（四）：动态代理
一.静态代理 在开始去学习反射实现的动态代理前,我们先需要了解代理设计模式,那何为代理呢? 代理模式: 为其他对象提供一种代理,以控制对这个对象的访问. 先看一张代理模式的结构图: 简单的理解代理设计 ...
- Reasoning and Learing学习笔记
Assignment 1 question 1 1.clisp安装及运行->参考博客 2.参数为列表,参考PPT 3.把嵌套列表变成非嵌套->题目P07 代码