ABC133F - Colorful Tree
ABC133FColorful Tree
题意
给定一颗边有颜色和权值的树,多次询问,每次询问,首先更改颜色为x的边的权值为y,然后输出u到v的距离。
数据都是1e5量级的。
思路
我自己一开始用树链剖分的做法。
// #pragma GCC optimize(2)
// #pragma GCC optimize(3)
// #pragma GCC optimize(4)
#include <algorithm>
#include <iterator>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <iomanip>
#include <bitset>
#include <cctype>
#include <cstdio>
#include <string>
#include <vector>
#include <stack>
#include <cmath>
#include <queue>
#include <list>
#include <map>
#include <set>
#include <cassert>
// #include<bits/extc++.h>
// using namespace __gnu_pbds;
using namespace std;
#define pb push_back
#define fi first
#define se second
#define debug(x) cerr<<#x << " := " << x << endl;
#define bug cerr<<"-----------------------"<<endl;
#define FOR(a, b, c) for(int a = b; a <= c; ++ a) typedef long long ll;
typedef long double ld;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll; const int inf = 0x3f3f3f3f;
const ll inff = 0x3f3f3f3f3f3f3f3f;
const int mod = ; template<typename T>
inline T read(T&x){
x=;int f=;char ch=getchar();
while (ch<''||ch>'') f|=(ch=='-'),ch=getchar();
while (ch>=''&&ch<='') x=x*+ch-'',ch=getchar();
return x=f?-x:x;
} /**********showtime************/
const int maxn = 1e5+;
vector<int>mp[maxn];
vector<int>col[maxn];
vector<int>wen[maxn];
struct E{
int u, v, c, d;
void init(int U, int V, int C, int D) {
u = U;
v = V;
c = C;
d = D;
}
} edge[maxn];
int dp[maxn], son[maxn], sz[maxn], pa[maxn];
void dfs1(int u, int fa) {
sz[u] = ;
pa[u] = fa;
dp[u] = dp[fa] + ;
int mx = ;
for(int v : mp[u]) {
if(v == fa) continue;
dfs1(v, u);
sz[u] += sz[v];
if(mx < sz[v]) mx = sz[v], son[u] = v;
}
}
int id[maxn], top[maxn], tot;
void dfs2(int u, int fa, int tp) {
top[u] = tp;
id[u] = ++tot; if(son[u]) dfs2(son[u], u, tp); for(int v : mp[u]) {
if(v == fa || v == son[u]) continue;
dfs2(v, u, v);
}
} ll ans[maxn];
ll sum[maxn<<]; void update(int pos, int val, int le, int ri, int rt) {
if(le == ri) {
sum[rt] = val;
return;
}
int mid = (le + ri) >> ;
if(pos <= mid) update(pos, val, le, mid, rt<<);
else update(pos, val, mid+, ri, rt<<|);
sum[rt] = sum[rt<<] + sum[rt<<|];
}
ll query(int L, int R, int le, int ri, int rt) {
if(le >= L && ri <= R) {
return sum[rt];
}
ll res = ;
int mid = (le + ri) >> ;
if(mid >= L) res += query(L, R, le, mid, rt<<);
if(mid < R) res += query(L, R, mid+, ri, rt<<|);
return res;
}
int n,m;
ll solve(int u, int v) {
ll res = ;
while(top[u] != top[v]) {
if(dp[top[u]] > dp[top[v]]) {
res += query(id[top[u]] , id[u], , n, );
u = pa[top[u]];
}
else {
res += query(id[top[v]], id[v], , n, );
v = pa[top[v]];
}
}
if(u != v) {
if(id[u] < id[v]) res += query(id[u]+, id[v],, n, );
else res += query(id[v]+, id[u], , n, );
}
return res;
}
struct Q{
int x, y;
int u,v;
}q[maxn];
int main(){
scanf("%d%d", &n, &m);
for(int i=; i<n; i++) {
int u, v, c, d;
scanf("%d%d%d%d", &u, &v, &c, &d);
edge[i].init(u, v, c, d);
mp[u].pb(v);
mp[v].pb(u);
col[c].pb(i);
} dfs1(, );
dfs2(, , ); for(int i=; i<=m; i++) {
scanf("%d%d%d%d", &q[i].x, &q[i].y, &q[i].u, &q[i].v);
wen[q[i].x].pb(i);
}
for(int i=; i<=n; i++) {
if(wen[i].size() == || col[i].size() == ) continue;
/// count
for(int dd : col[i]) {
int u = edge[dd].u;
int v = edge[dd].v;
if(dp[u] < dp[v]) swap(u, v);
update(id[u], , , n, );
}
for(int dd: wen[i]) {
ans[dd] += solve(q[dd].u, q[dd].v) * q[dd].y;
}
for(int dd : col[i]) {
int u = edge[dd].u;
int v = edge[dd].v;
if(dp[u] < dp[v]) swap(u, v);
update(id[u], , , n, );
} ///sum
for(int dd : col[i]) {
int u = edge[dd].u;
int v = edge[dd].v;
if(dp[u] < dp[v]) swap(u, v);
update(id[u], edge[dd].d, , n, );
}
for(int dd: wen[i]) {
ans[dd] -= solve(q[dd].u, q[dd].v);
}
for(int dd : col[i]) {
int u = edge[dd].u;
int v = edge[dd].v;
if(dp[u] < dp[v]) swap(u, v);
update(id[u], , , n, );
}
} for(int i=; i<n; i++) {
int u = edge[i].u;
int v = edge[i].v;
if(dp[u] < dp[v]) swap(u, v);
update(id[u], edge[i].d, , n, );
}
for(int i=; i<=m; i++) {
int u = q[i].u;
int v = q[i].v;
ans[i] += solve(u, v);
}
for(int i=; i<=m; i++) printf("%lld\n", ans[i]);
return ;
}
还有树上莫队和树上差分的思路。
正在搞树上莫队的方法。搞好了。
// #pragma GCC optimize(2)
// #pragma GCC optimize(3)
// #pragma GCC optimize(4)
#include <algorithm>
#include <iterator>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <iomanip>
#include <bitset>
#include <cctype>
#include <cstdio>
#include <string>
#include <vector>
#include <stack>
#include <cmath>
#include <queue>
#include <list>
#include <map>
#include <set>
#include <cassert>
// #include<bits/extc++.h>
// using namespace __gnu_pbds;
using namespace std;
#define pb push_back
#define fi first
#define se second
#define debug(x) cerr<<#x << " := " << x << endl;
#define bug cerr<<"-----------------------"<<endl;
#define FOR(a, b, c) for(int a = b; a <= c; ++ a) typedef long long ll;
typedef long double ld;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll; const int inf = 0x3f3f3f3f;
const ll inff = 0x3f3f3f3f3f3f3f3f;
const int mod = ; template<typename T>
inline T read(T&x){
x=;int f=;char ch=getchar();
while (ch<''||ch>'') f|=(ch=='-'),ch=getchar();
while (ch>=''&&ch<='') x=x*+ch-'',ch=getchar();
return x=f?-x:x;
} /**********showtime************/
const int maxn = 1e5+;
vector<int>mp[maxn];
struct E {
int u,v;
int c,d;
void init(int U, int V ,int C, int D) {
u = U; v = V; c = C; d = D;
}
} edge[maxn];
int dp[maxn];
int st[maxn ],ed[maxn], tim = ;
int fa[maxn][];
int a[maxn*];
void dfs(int u, int o) { dp[u] = dp[o] + ;
st[u] = ++tim;
a[tim] = u;
fa[u][] = o;
for(int i=; i<; i++)
fa[u][i] = fa[fa[u][i-]][i-];
for(int v : mp[u]) {
if(v == o) continue;
dfs(v, u);
}
ed[u] = ++tim;
a[tim] = u;
}
int lca(int u, int v) {
if(dp[u] < dp[v]) swap(u, v);
for(int i=; i>=; i--) {
if(dp[fa[u][i]] >= dp[v])
u = fa[u][i];
}
if(u == v) return u;
for(int i=; i>=; i--) {
if(fa[u][i] != fa[v][i])
u = fa[u][i], v = fa[v][i];
}
return fa[u][];
}
int c[maxn], d[maxn];
struct Q{
// int u, v;
int le, ri;
int lb;
int x, y;
int id;
} qry[maxn];
bool cmp(Q a, Q b) {
if(a.lb != b.lb)
return a.lb < b.lb;
return a.ri < b.ri;
}
int ans[maxn];
int sum,colcnt[maxn],colsum[maxn],pot[maxn];
void add(int id) { if(pot[a[id]] == ) pot[a[id]] = ;
else pot[a[id]] = ;
int col = c[a[id]];
if(pot[a[id]]) {
colcnt[col] ++;
colsum[col] += d[a[id]];
sum += d[a[id]];
}
else {
colcnt[col] --;
colsum[col] -= d[a[id]];
sum -= d[a[id]];
}
}
int main(){
int n,m;
scanf("%d%d", &n, &m);
int block = ;
for(int i=; i<n; i++) {
int u,v,c,d;
scanf("%d%d%d%d", &u, &v, &c, &d);
edge[i].init(u, v, c, d);
mp[u].pb(v);
mp[v].pb(u);
}
dfs(, );
for(int i=; i<n; i++) {
int u = edge[i].u, v = edge[i].v;
if(dp[u] < dp[v]) swap(u, v);
c[u] = edge[i].c;
d[u] = edge[i].d;
} for(int i=; i<=m; i++) {
int x, y, u, v;
scanf("%d%d%d%d", &x, &y, &u, &v);
qry[i].id = i;
qry[i].x = x;
qry[i].y = y;
if(st[u] > st[v]) swap(u, v);
int _lca = lca(u, v);
if(_lca == u) {
qry[i].le = st[u]+;
qry[i].ri = st[v];
qry[i].lb = qry[i].le / block;
}
else {
qry[i].le = ed[u];
qry[i].ri = st[v];
qry[i].lb = qry[i].le / block;
}
} sort(qry+, qry++m, cmp);
int le = , ri = ;
for(int i=; i<=m; i++) {
while(le < qry[i].le) add(le), le++;
while(le > qry[i].le) le--, add(le);
while(ri < qry[i].ri) ri++, add(ri);
while(ri > qry[i].ri) add(ri), ri--; ans[qry[i].id] = sum - colsum[qry[i].x] + colcnt[qry[i].x] * qry[i].y;
}
for(int i=; i<=m; i++) printf("%d\n", ans[i]);
return ;
}
ABC133F - Colorful Tree的更多相关文章
- @atcoder - ABC133F@ Colorful Tree
目录 @description@ @solution - 1@ @accepted code - 1@ @solution - 2@ @accepted code - 2@ @details@ @de ...
- 2017 Multi-University Training Contest - Team 1 1003&&HDU 6035 Colorful Tree【树形dp】
Colorful Tree Time Limit: 6000/3000 MS (Java/Others) Memory Limit: 131072/131072 K (Java/Others)T ...
- hdu6035 Colorful Tree 树形dp 给定一棵树,每个节点有一个颜色值。定义每条路径的值为经过的节点的不同颜色数。求所有路径的值和。
/** 题目:hdu6035 Colorful Tree 链接:http://acm.hdu.edu.cn/showproblem.php?pid=6035 题意:给定一棵树,每个节点有一个颜色值.定 ...
- AtCoder Beginner Contest 133 F Colorful Tree
Colorful Tree 思路: 如果强制在线的化可以用树链剖分. 但这道题不强制在线,那么就可以将询问进行差分,最后dfs时再计算每个答案的修改值, 只要维护两个数组就可以了,分别表示根节点到当前 ...
- HDU 6035 - Colorful Tree | 2017 Multi-University Training Contest 1
/* HDU 6035 - Colorful Tree [ DFS,分块 ] 题意: n个节点的树,每个节点有一种颜色(1~n),一条路径的权值是这条路上不同的颜色的数量,问所有路径(n*(n-1)/ ...
- [HDU6793] Tokitsukaze and Colorful Tree
题目 又是一个条历新年,窗前的灼之花又盛开了. 时隔多年,现在只有这一棵树上盛开着残存的 \(n\) 朵灼之花了. 尽管如此,这些灼之 花仍散发出不同色彩的微弱的光芒. 灼之花的生命极为短暂,但它的花 ...
- Colorful tree
cnbb 我被数组清零卡了一天.. 子树改色询问子树颜色数.. 先考虑颜色为x的节点对祖先答案的贡献,那么我们考虑把所有这些节点都搞出来,按dfs序排序,然后考虑每个节点a掌管的祖先是它和按dfs序的 ...
- 2017ACM暑期多校联合训练 - Team 1 1003 HDU 6035 Colorful Tree (dfs)
题目链接 Problem Description There is a tree with n nodes, each of which has a type of color represented ...
- HDU-6035:Colorful Tree(虚树+DP)
这里有三道长得像的题: 一:HDU6036: There is a tree with nn nodes, each of which has a type of color represented ...
随机推荐
- ES 22 - Elasticsearch中如何进行日期(数值)范围查询
目录 1 范围查询的符号 2 数值范围查询 3 时间范围查询 3.1 简单查询示例 3.2 关于时间的数学表达式(date-math) 3.3 关于时间的四舍五入 4 日期格式化范围查询(format ...
- chapter02 - 03
1.分别用cat \tac\nl三个命令查看文件/etc/ssh/sshd_config文件中的内容,并用自己的话总计出这三个文档操作命令的不同之处? 答:cat /etc/ssh/sshd_conf ...
- 关于JS、JQ首行首列固定展示
当时遇见这个问题 是医院手麻系统大批量数据展示,由于是旧项目系统没有使用到前端的架构 只能使用JQ,JS, css完成 也谢谢给予我支持的同行们 固定首行数据: 采用函数的方式进行 JQ /** * ...
- 【MySQL】Illegal mix of collations (utf8mb4_general_ci,IMPLICIT) and ...
线上遇到这个问题,详细信息如下: SQL state [HY000]; error code [1267]; Illegal mix of collations (utf8mb4_general_ci ...
- 【iOS】UILabel 常用属性设置
UILabel 的一些常用属性,示例代码如下: // 字体大小 label.font = [UIFont systemFontOfSize:14.0]; label.font = [UIFont fo ...
- nginx 之负载均衡 :PHP session 跨多台服务器配置
公司一个项目单点压力越来越大,考虑到稳定性和降压,使用nginx做负载均衡,将请求分发到多个docker上去,这里记录下PHP多服务器间的会话session共享问题,解决方案是把session单独存在 ...
- Ubuntu中修改默认开机项
1首先,按住Ctrl+Alt+t打开终端 2输入cd /etc/default 3输入sudo sudo nano grub 并按照提示输入密码 4在我们开机的时候,可以看到自己想要默认的开机项是多少 ...
- RocketMQ中Broker的刷盘源码分析
上一篇博客的最后简单提了下CommitLog的刷盘 [RocketMQ中Broker的消息存储源码分析] (这篇博客和上一篇有很大的联系) Broker的CommitLog刷盘会启动一个线程,不停地 ...
- React 如何搭建脚手架
React 如何搭建脚手架 npm install -g create-react-app //安装 create-react-app react-demo // react-demo ...
- 【Java例题】5.1 多项式计算
1. 计算下列多项式的值. pn=an*x^n+...+a1*x+a0其中,"^"表示乘方. x.n以及ai(i=0,1,...,n-1)由键盘输入. package chapte ...