树链剖分:大佬的博客 大神的博客
题目:【模板】树链剖分
学长的板子:
#include <bits/stdc++.h>
using namespace std;
const int N = 2e5+10;
int mod = 1e9+7;
int n, m, r, p;
template<typename T>
struct Heavy_Light_Decomposition{
//head u的子节点编号,to 子节点, nx 下一子节点的编号
int head[N], to[N], nx[N], tot, cnt;
//dep 深度,fa 父亲 ,sz 子树大小 ,son 儿子 ,w权值
int dep[N], fa[N], sz[N], son[N], w[N];
//top 重儿子的祖先 ,id 刨分后的编号 ,wt 刨分后的权值
int top[N], id[N], wt[N];
//val 维护区间和 ,lz lazy标记
T val[N<<2], lz[N<<2];
void init(int n){
tot = cnt = 0;
for(int i = 0; i <= n; ++i) head[i] = -1;
}
void add(int u, int v){
to[tot] = v;
nx[tot] = head[u];
head[u] = tot++;
}
void dfs1(int u,int f){//初始化dep son sz fa,找出重儿子
dep[u] = dep[f] + 1; fa[u] = f; sz[u] = 1;
son[u] = 0;
for(int i = head[u]; ~i; i = nx[i]){
int v = to[i];
if(v == f) continue;
dfs1(v, u);
sz[u] += sz[v];
if(sz[v] > sz[son[u]]) son[u] = v;
}
}
void dfs2(int u,int topf){//树刨 初始化id wt top
id[u] = ++cnt;
wt[cnt] = w[u];
top[u] = topf;
if(!son[u]) return;
dfs2(son[u], topf);
for(int i = head[u]; ~i; i = nx[i]){
int v = to[i];
if(v == fa[u] || v == son[u]) continue;
dfs2(v, v);
}
}
void pushdown(int rt,int l,int r){
if(!lz[rt]) return ;
int mid = (l+r) >> 1;
lz[rt<<1] += lz[rt]; lz[rt<<1] %= mod;
lz[rt<<1|1] += lz[rt]; lz[rt<<1|1] %= mod;
val[rt<<1] += lz[rt] * (mid-l+1) %mod; val[rt<<1] %= mod;
val[rt<<1|1] += lz[rt] * (r-mid) %mod; val[rt<<1|1] %=mod;
lz[rt] = 0;
}
void pushup(int rt){
val[rt] = val[rt<<1] + val[rt<<1|1];
val[rt] %= mod;
}
void build(int rt, int l, int r){
if(l == r){
val[rt] = wt[l];
val[rt] %= mod;
return ;
}
int mid = (l+r) >> 1;
build(rt<<1, l, mid);
build(rt<<1|1, mid+1, r);
pushup(rt);
}
T query(int rt, int l, int r, int L, int R){
if(L <= l && R >= r){
return val[rt];
}
int mid = (l+r) >> 1; T ans = 0;
pushdown(rt, l, r);
if(L <= mid) ans += query(rt<<1, l, mid, L, R), ans %= mod;
if(R > mid) ans += query(rt<<1|1, mid+1, r, L, R), ans %= mod;
pushup(rt);
return ans;
}
T queryRange(int l, int r){//查询l,r的最短路路径和
T ans = 0;
while(top[l] != top[r]){
if(dep[top[l]] < dep[top[r]]) swap(l, r);
ans += query(1, 1, n, id[top[l]], id[l]);
ans %= mod;
l = fa[top[l]];
}
if(dep[l]>dep[r]) swap(l, r);
ans += query(1, 1, n, id[l], id[r]);
return ans%mod;
}
void update(int rt, int l, int r, int L, int R, int x){
if(L <= l &&R >= r){
lz[rt] += x; lz[rt] %=mod;
val[rt] += 1LL*x*(r-l+1)%mod;
val[rt] %=mod;
return ;
}
int mid = (l+r)>>1;
pushdown(rt, l, r);
if(L <= mid) update(rt<<1, l, mid, L, R, x);
if(R > mid) update(rt<<1|1, mid+1, r, L, R, x);
pushup(rt);
}
//修改l,r的最短路径的权值
void updateRange(int l, int r, int x){
while(top[l] != top[r]){
if(dep[top[l]] < dep[top[r]]) swap(l, r);
update(1, 1, n, id[top[l]], id[l], x);
l = fa[top[l]];
}
if(dep[l]>dep[r]) swap(l, r);
update(1, 1, n, id[l], id[r], x);
}
//修改子树权值
void updateSon(int x,int y){
update(1, 1, n, id[x], id[x]+sz[x]-1, y);
}
//查询子树和
T querySon(int x){
return query(1, 1, n, id[x], id[x]+sz[x]-1);
}
};
Heavy_Light_Decomposition<long long>hld;//模板
int main(){
//n个点,m次操作,r是根节点,p模
scanf("%d%d%d%d", &n, &m, &r, &p);
hld.init(n);
mod = p;
for(int i = 1; i <= n; ++i){
scanf("%d", &hld.w[i]);
}
for(int i = 1; i < n; ++i){
int u, v; scanf("%d%d", &u, &v);
hld.add(u, v);
hld.add(v, u);
}
hld.dfs1(r, 0);
hld.dfs2(r, r);
hld.build(1, 1, n);
for(int i = 1; i <= m; ++i){
int id, x, y, z;
scanf("%d", &id);
if(id == 1){
scanf("%d%d%d", &x, &y, &z);
hld.updateRange(x, y, z);
}
else if(id == 2){
scanf("%d%d", &x, &y);
printf("%lld\n", hld.queryRange(x, y));
}
else if(id == 3){
scanf("%d%d", &x, &y);
hld.updateSon(x, y);
}
else scanf("%d", &x), printf("%lld\n", hld.querySon(x));
}
return 0;
}
A - Aragorn’s Story
题解:板子题(awsl,lazy标记忘记初始化了…所以总是不过
用线段树做总是不对…用树状数组做了一遍
#pragma comment(linker, "/STACK:10240000,10240000")
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 1e6+10;
int n, m, r, p;
template<typename T>
struct Heavy_Light_Decomposition{
//head u的子节点编号,to 子节点, nx 下一子节点的编号
int head[N], to[N], nx[N], tot, cnt;
//dep 深度,fa 父亲 ,sz 子树大小 ,son 儿子 ,w权值
int dep[N], fa[N], sz[N], son[N], w[N];
//top 重儿子的祖先 ,id 刨分后的编号 ,wt 刨分后的权值
int top[N], id[N], wt[N],tree[N];
//val 维护区间和 ,lz lazy标记
T val[N<<2], lz[N<<2];
void init(int n){
tot = cnt = 0;
for(int i = 0; i <= n; ++i) head[i] = -1;
}
void add(int u, int v){
to[tot] = v;
nx[tot] = head[u];
head[u] = tot++;
}
void dfs1(int u,int f){//初始化dep son sz fa,找出重儿子
dep[u] = dep[f] + 1; fa[u] = f; sz[u] = 1;
son[u] = 0;
for(int i = head[u]; ~i; i = nx[i]){
int v = to[i];
if(v == f) continue;
dfs1(v, u);
sz[u] += sz[v];
if(sz[v] > sz[son[u]]) son[u] = v;
}
}
void dfs2(int u,int topf){//树刨 初始化id wt top
id[u] = ++cnt;
wt[cnt] = w[u];
top[u] = topf;
if(!son[u]) return;
dfs2(son[u], topf);
for(int i = head[u]; ~i; i = nx[i]){
int v = to[i];
if(v == fa[u] || v == son[u]) continue;
dfs2(v, v);
}
}
int lowbit(int x)
{
return x&(-x);
}
void ADD(int p,int v)
{
for(int i=p;i<=n;i+=lowbit(i))
tree[i]+=v;
}
int SUM(int p)
{
int ret=0;
for(int i=p;i;i-=lowbit(i))
ret+=tree[i];
return ret;
}
void Change(int u,int v,int K)
{
int f1=top[u],f2=top[v];
while(f1!=f2)
{
if(dep[f1]<dep[f2])
{
swap(f1,f2);
swap(u,v);
}
ADD(id[f1],K);
ADD(id[u]+1,-K);
u=fa[f1];
f1=top[u];
}
if(dep[u]<dep[v]) swap(u,v);
ADD(id[v],K);
ADD(id[u]+1,-K);
}
};
Heavy_Light_Decomposition<int>hld;//模板
int main(){
while(scanf("%d%d%d", &n, &p, &m)!=EOF)
{
r=1;
hld.init(n);
for(int i = 1; i <= n; ++i){
scanf("%d", &hld.w[i]);
}
for(int i = 1; i <= p; ++i){
int u, v; scanf("%d%d", &u, &v);
hld.add(u, v);
hld.add(v, u);
}
hld.dfs1(r, 0);
hld.dfs2(r, r);
memset(hld.tree,0,sizeof(hld.tree));
for(int i=1;i<=n;i++)
{
hld.ADD(hld.id[i],hld.w[i]);
hld.ADD(hld.id[i]+1,-hld.w[i]);
}
for(int i = 1; i <= m; ++i){
int id, x, y, z;
char str[5];
//getchar();
scanf("%s", str);
if(str[0] == 'I'){
scanf("%d%d%d", &x, &y, &z);
hld.Change(x, y, z);
}
else if(str[0] == 'Q'){
scanf("%d", &x);
printf("%d\n", hld.SUM(hld.id[x]));
}
else if(str[0] == 'D'){
scanf("%d%d%d", &x, &y, &z);
hld.Change(x, y, -z);
}
}
}
}
Housewife Wind
题解:树刨,边化点
模板开太大…没有意识到,然后就疯狂的TTTTT,以及不要用
v
e
c
t
o
r
vector
vector,慢的要死
#include <cstdio>
#include <algorithm>
using namespace std;
const int N = 2e5+10;
int mod = 1e9+7;
int n, m, r, p;
template<typename T>
struct Heavy_Light_Decomposition{
//head u的子节点编号,to 子节点, nx 下一子节点的编号
int head[N], to[N], nx[N], tot, cnt;
//dep 深度,fa 父亲 ,sz 子树大小 ,son 儿子 ,w权值
int dep[N], fa[N], sz[N], son[N], w[N];
//top 重儿子的祖先 ,id 刨分后的编号 ,wt 刨分后的权值
int top[N], id[N], wt[N];
//val 维护区间和 ,lz lazy标记
T val[N<<2], lz[N<<2];
void init(int n){
tot = cnt = 0;
for(int i = 0; i <= n; ++i) head[i] = -1;
}
void add(int u, int v){
to[tot] = v;
nx[tot] = head[u];
head[u] = tot++;
}
void dfs1(int u,int f){//初始化dep son sz fa,找出重儿子
dep[u] = dep[f] + 1; fa[u] = f; sz[u] = 1;
for(int i = head[u]; ~i; i = nx[i]){
int v = to[i];
if(v == f) continue;
dfs1(v, u);
sz[u] += sz[v];
if(sz[v] > sz[son[u]]) son[u] = v;
}
}
void dfs2(int u,int topf){//树刨 初始化id wt top
id[u] = ++cnt;
wt[cnt] = w[u];
top[u] = topf;
if(!son[u]) return;
dfs2(son[u], topf);
for(int i = head[u]; ~i; i = nx[i]){
int v = to[i];
if(v == fa[u] || v == son[u]) continue;
dfs2(v, v);
}
}
void pushup(int rt){
val[rt] = val[rt<<1] + val[rt<<1|1];
val[rt] %= mod;
}
void build(int rt, int l, int r){
if(l == r){
val[rt] = wt[l];
val[rt] %= mod;
return ;
}
int mid = (l+r) >> 1;
build(rt<<1, l, mid);
build(rt<<1|1, mid+1, r);
pushup(rt);
}
T query(int rt, int l, int r, int L, int R){
if(L <= l && R >= r){
return val[rt];
}
int mid = (l+r) >> 1; T ans = 0;
//pushdown(rt, l, r);
if(L <= mid) ans += query(rt<<1, l, mid, L, R), ans %= mod;
if(R > mid) ans += query(rt<<1|1, mid+1, r, L, R), ans %= mod;
pushup(rt);
return ans;
}
T queryRange(int l, int r){//查询l,r的最短路路径和
T ans = 0;
while(top[l] != top[r]){
if(dep[top[l]] < dep[top[r]]) swap(l, r);
ans += query(1, 1, n, id[top[l]], id[l]);
ans %= mod;
l = fa[top[l]];
}
/********边化为点***********/
if(l == r) return ans;///特判不移动
if(dep[l]>dep[r]) swap(l, r);
///由于边变成点,改为id[son[l]]
ans += query(1, 1, n, id[son[l]], id[r]);
return ans%mod;
}
void update(int rt, int l, int r, int L, int R, int x){
if(L <= l &&R >= r){
//lz[rt] += x; lz[rt] %=mod;
val[rt] = 1LL*x%mod;
val[rt] %=mod;
return ;
}
int mid = (l+r)>>1;
//pushdown(rt, l, r);
if(L <= mid) update(rt<<1, l, mid, L, R, x);
if(R > mid) update(rt<<1|1, mid+1, r, L, R, x);
pushup(rt);
}
//修改l,r的最短路径的权值
void updateRange(int l, int r, int x){
while(top[l] != top[r]){
if(dep[top[l]] < dep[top[r]]) swap(l, r);
update(1, 1, n, id[top[l]], id[l], x);
l = fa[top[l]];
}
if(dep[l]>dep[r]) swap(l, r);
update(1, 1, n, id[l], id[r], x);
}
};
struct note
{
int u,v,w;
}h[N];
Heavy_Light_Decomposition<int>hld;//模板
int main(){
scanf("%d%d%d", &n, &m, &p);
r=1;
hld.init(n);
for(int i = 1; i < n; ++i){
int u, v,w; scanf("%d%d%d", &u, &v,&w);
h[i].u=u,h[i].v=v,h[i].w=w;
hld.add(u, v);
hld.add(v, u);
}
hld.dfs1(r, 0);
hld.dfs2(r, r);
hld.build(1, 1, n);
int xx;
for(int i=1;i<n;i++){///比较一条边上哪个点的深度大,就把权值加在哪个点上
if(hld.dep[h[i].u]<hld.dep[h[i].v]) xx=h[i].v;
else xx=h[i].u;
hld.updateRange(xx,xx,h[i].w);
}
for(int i = 1; i <= m; ++i){
int id, x, y, z;
//getchar();
scanf("%d", &id);
if(id == 1){
scanf("%d%d", &x, &y);
if(hld.dep[h[x].u]<hld.dep[h[x].v]) xx=h[x].v;
else xx=h[x].u;
hld.updateRange(xx,xx,y);
}
else if(id == 0){
scanf("%d", &x);
printf("%d\n", hld.queryRange(x, p));
p=x;
}
}
return 0;
}
B - Valid Sets
题意:
给定一棵树,任意选定节点,节点间是联通的(也可以是同一个节点),使这些个点的路径上的点中最大值和最小值小于
d
d
d
题解:树形
d
p
dp
dp
枚举节点作为最大节点,然后遍历它所控制的范围计算符合条件的方案数
#include<bits/stdc++.h>
#define mp make_pair
#define maxn 10000000
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, int> pli;
typedef pair<ll, ll> pll;
typedef long double ld;
const int N=1e6+10;
const int MAXN=20010;
const int INF=0x3f3f3f3f;
const double eps=0.0000001;
const ll mod=998244353 ;
int n,m,x,y,k,cnt;
int d,a[maxn],root;
const int MOD=1e9+7;
vector<int>G[maxn];
long long dfs(int u,int fa)
{
long long res=1;
int v;
for(int i=0;i<(int)G[u].size();i++)
{
v=G[u][i];
if(v==fa)///单向
continue;
if(a[v]>a[root])///使得root最大,不需要考虑最小值
continue;
if(a[v]==a[root]&&v>root)///避免重复
continue;
if(a[root]-a[v]>d)///最小的节点都满足的话其他值都满足
continue;
res=res*(dfs(v,u)+1)%MOD;///计算子树的数目
}
return res;
}
int main()
{
ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
scanf("%d%d",&d,&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<=n-1;i++)
{
int x,y;
scanf("%d%d",&x,&y);
G[x].push_back(y);
G[y].push_back(x);
}
long long ans=0;
///枚举每个结点,把每个节点都当作最大节点来进行计算
for(int i=1;i<=n;i++)
{
root=i;
ans=(ans+dfs(root,-1))%MOD;
}
printf("%I64d\n",ans);
//system("pause");
return 0;
}
D. Choosing Capital for Treeland
题意:
这道题给出n个节点的树,其中的边是有向边,让你从其中选择一个节点,使得从该节点出发能够到达其他所有节点,中途可以改变边的方向,但需要使改变的次数最少。
题解:树形
d
p
dp
dp
先预处理一个点
u
u
u,将
u
u
u 点作为根进行遍历,记录反向的个数
w
w
w
会发现一个很有趣的规律;
当将其他结点v 作为根时的反向边的个数为
n
n
n
n
=
w
−
2
∗
(
u
到
v
路
径
上
的
反
向
边
)
+
(
u
到
v
路
径
长
度
)
n=w -2*(u到v路径上的反向边)+ (u到v路径长度)
n=w−2∗(u到v路径上的反向边)+(u到v路径长度)
然后找出所有节点的最小值就可以了
#include<bits/stdc++.h>
#define mp make_pair
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, int> pli;
typedef pair<ll, ll> pll;
typedef long double ld;
const int N=1e6+10;
const int MAXN=20010;
const int INF=0x3f3f3f3f;
const double eps=0.0000001;
const ll mod=998244353 ;
int n,m,x,y,k,cnt;
struct note
{
int u,v,w;
}h;
vector<note>E[N];
int dp[N],dep[N],ans[N],pop=INF;
void dfs1(int x,int fa)
{
dep[x]=dep[fa]+1;
for(int i=0;i<E[x].size();i++)
{
note y;
y=E[x][i];
if(y.v==fa) continue;
if(y.w<0) cnt++;
dp[y.v]=dp[y.u]+((y.w==-1)?1:0);///被三元符坑了忘记加括号...
dfs1(y.v,y.u);
}
}
int main()
{
ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
scanf("%d", &n);
for(int i=0;i<n-1;i++)
{
int u,v;
scanf("%d%d",&u,&v);
h.u=u,h.v=v,h.w=1;
E[u].push_back(h);
h.u=v,h.v=u,h.w=-1;
E[v].push_back(h);
}
dep[0]=-1;
dfs1(1,0);
ans[1]=cnt;
pop=min(pop,ans[1]);
for(int i=2;i<=n;i++)
{
ans[i]=ans[1]-2*dp[i]+dep[i];
pop=min(pop,ans[i]);
//printf("%d\n",dp[i]);
}
printf("%d\n",pop);
for(int i=1;i<=n;i++)
{
if(pop==ans[i])
printf("%d ",i);
}
}