【csp模拟赛6】树上统计-启发式合并,线段树合并
2024-09-02 10:30:05
30%:暴力
40%:枚举L,R从L~n枚举,R每增大一个,更新需要的边(bfs实现)60%:枚举每条边,
计算每条边的贡献另外20%的数据:枚举每条边,计算每条边的贡献100%:对于每一条边统计
有多少个区间跨过这条边即可统计这一问题的对偶问题,有多少个区间没跨过会更方便使用启发式合并+
并查集统计子树内的,使用启发式合并+set统计子树外的
代码:
#include<cstdio>
#include<cstdlib>
#include<set>
#include<vector>
#include<iostream>
#define LL long long
#define int long long
using namespace std;
const int MAXN = 1e5 + 10;
inline int read() {
char c = getchar(); int x = 0, f = 1;
while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
return x * f;
}
int N, siz[MAXN], son[MAXN], dsu[MAXN], Son, vis[MAXN], ds[MAXN];
vector<int> v[MAXN];
set<int> s;
LL outt, inn, ans;
void dfs(int x, int _fa)
{
siz[x] = 1;
for(int i = 0; i < v[x].size(); i++) {
int to = v[x][i]; if(to == _fa) continue;
dfs(to, x);
siz[x] += siz[to];
if(siz[to] > siz[son[x]]) son[x] = to;
}
}
LL calc(LL x) {
return x * (x - 1) / 2;
}
void Clear() {
s.clear();
outt = calc(N);
inn = 0;
s.insert(0);
s.insert(N + 1);
}
int find(int x)
{
return dsu[x] == x ? dsu[x] : dsu[x] = find(dsu[x]);
}
void solve(int x)
{
s.insert(x);
set<int>::iterator s1, s2, it;
s1 = s2 = it = s.find(x);
s1--; s2++;
outt -= calc((*s2) - (*s1) - 1);
outt += calc((*s2) - (*it) - 1) + calc((*it) - (*s1) - 1);
vis[x] = 1;
if(vis[x - 1]) {
int fx = find(x - 1), fy = find(x);
inn += ds[fx] * ds[fy];
dsu[fx] = fy;
ds[fy] += ds[fx];
}
if(vis[x + 1]) {
int fx = find(x + 1), fy = find(x);
inn += ds[fx] * ds[fy];
dsu[fx] = fy;
ds[fy] += ds[fx];
}
}
void Add(int x, int fa)
{
solve(x);
for(int i = 0; i < v[x].size(); i++)
{
int to = v[x][i];
if(to == fa || to == Son) continue;
Add(to, x);
}
}
void Delet(int x, int fa) {
vis[x] = 0; ds[x] = 1; dsu[x] = x;
for(int i = 0; i < v[x].size(); i++) {
int to = v[x][i];
if(to == fa) continue;
Delet(to, x);
}
}
void dfs2(int x, int fa, int opt)
{
for(int i = 0; i < v[x].size(); i++)
{
int to = v[x][i];
if(to == fa || (to == son[x])) continue;
dfs2(to, x, 0);
}
if(son[x]) dfs2(son[x], x, 1); Son = son[x];
Add(x, fa);
ans += calc(N) - inn - outt;
if(opt == 0) Delet(x, fa), Clear(), Son = 0;
}
signed main()
{
#ifdef yilnr
#else
freopen("treecnt.in","r",stdin);
freopen("treecnt.out","w",stdout);
#endif
N = read();
for(int i = 1; i <= N - 1; i++)
{
int x = read(), y = read();
v[x].push_back(y);
v[y].push_back(x);
}
for(int i = 1; i <= N; i++) dsu[i] = i,ds[i] = 1;
dfs(1,0);
Clear();
dfs2(1,0,0);
printf("%lld\n",ans);
return 0;
}
/*
4
1 4
1 3
2 4
*/
最新文章
- 关于C#的委托(delegate)的自我理解
- 如何在IIS7/7.5上配置IISADMPWD
- SSH框架搭建详解 及 乱码处理
- Struts2的模板和主题theme及自定义theme的使用
- js-特效部分学习-offsetParent、scrollHeight 、动画函数的封装
- iOS中dyld缓存的实现原理是怎样的?
- C#将DataTable导出Execl、Word、Xml
- Eclipse使用git最简易流程
- Pi
- 安卓弹出对话框——AlertDialog(二)
- Objective-C 学习 (二):Objective-C 实战之Hello World
- JS获取登录者IP和登录城市
- 99%的Linux运维工程师必须要掌握的命令及运用
- ORACLE数据库管理员的职责
- 分布式系统监视zabbix讲解二之邮件报警通知--技术流ken
- libSVM在matlab下的使用安装
- oracle partition table 分区表详解
- Android 各个版本新特性
- Django的学习(五)————实战问题
- springcloud 入门 1 (浅谈版本关系)