CodeForces - 468D Tree

先推一下第一个答案的公式:

a n s = Σ d e p [ i ] + d e p [ p [ i ] ] − 2 ∗ d e p [ l c a ( i , p [ i ] ) ] ans=Σ dep[i]+dep[p[i]]-2*dep[lca(i, p[i])] ans=Σdep[i]+dep[p[i]]2dep[lca(i,p[i])]
         = ( 2 ∗ Σ d e p [ i ] ) − 2 ∗ Σ d e p [ l c a ( i , p [ i ] ) ] =(2*Σdep[i])-2*Σdep[lca(i,p[i])] =(2Σdep[i])2Σdep[lca(i,p[i])]

所以我们要使 l c a lca lca d e p dep dep尽量小。

然后我们说说为什么要选取重心作为根,为什么后面这一坨可以取0。

感性证明一波。 我们知道,如果以重心作为根有一个性质: s i z [ i ] < = ⌊ n / 2 ⌋ siz[i]<= \lfloor{n/2} \rfloor siz[i]<=n/2,其中i为子树。(好像是??

那我们选取一个siz最大的子树来分类讨论。

  1. s i z = = n / 2 ( n 为 奇 ) siz==n/2(n为奇) siz==n/2(n) 我们将剩余子树内节点依次与此子树配对(这肯定能保证是排列),剩一个再和根配对就行了。
  2. s i z = = n / 2 ( n 为 偶 ) siz==n/2(n为偶) siz==n/2(n) 我们恰好将各个子树配对,最后根节点和自己在一起~~~
  3. s i z < n / 2 siz<n/2 siz<n/2 我们随便各个子树配对,最后总能转化成前面的情况。

bibi了半天感觉并没有什么用。

然后我们发现其实每个点会出现两次。(分别在 i i i p i pi pi中)定义在 i i i中的为 o u t out out,在 p i pi pi中为 i n in in

设子树 i i i中有 i n [ i ] in[i] in[i]个未匹配的入点, o u t [ i ] out[i] out[i]个未匹配出点。

根据定义得, o u t [ i ] < = ( Σ i n [ j ] ) − i n [ i ] out[i]<=(Σin[j])-in[i] out[i]<=(Σin[j])in[i]

                       i n [ i ] + o u t [ i ] < = Σ i n [ j ] in[i]+out[i]<=Σin[j] in[i]+out[i]<=Σin[j]

我们又知道 Σ i n [ i ] = = Σ o u t [ i ] = = n − j + 1 Σin[i]==Σout[i]==n-j+1 Σin[i]==Σout[i]==nj+1 j j j从1到n遍历取出点的第 j j j个点)

所以原式化为: i n [ i ] + o u t [ i ] < = n − j + 1 in[i]+out[i]<=n-j+1 in[i]+out[i]<=nj+1

women知道,每个独立子树是不能有在一起的(好凄情的故事 )。于是 i n [ i ] + o u t [ i ] in[i]+out[i] in[i]+out[i]被赋予了别样的含义:代表着还要匹配多少对。如果有 i n [ i ] + o u t [ i ] = = n − j + 1 in[i]+out[i]==n-j+1 in[i]+out[i]==nj+1,这就意味着第j个点要么是第 i i i棵子树中的点,要么与第 i i i棵子树中的点在一起。其余的就选最小的就行了。

最后的话:上述情况有些并不包括根,根需要另外安排。

#include<set>
#include<cstdio>
#include<iostream>
using namespace std;
typedef pair <int, int> Pair;

const int N = 1e5 + 2;
long long ans, dis[N];
int res[N], be[N], num[N], val[N << 1], block, n, head[N], dot[N << 1], nxt[N << 1], cnt, maxx[N], siz[N], root;
set <int> s[N];
set <Pair> a, b;

void addEdge(const int u, const int v, const int w) {
	dot[++ cnt] = v;
	nxt[cnt] = head[u];
	val[cnt] = w;
	head[u] = cnt;
}

void makeRoot(const int x, const int ba) {
	siz[x] = 1;
	for(int i = head[x]; i; i = nxt[i]) {
		int v = dot[i];
		if(v == ba) continue;
		makeRoot(v, x);
		siz[x] += siz[v];
		maxx[x] = max(maxx[x], siz[v]);
	}
	maxx[x] = max(maxx[x], n - siz[x]);
	if(maxx[root] > maxx[x]) root = x;
}

void init(const int x, const int ba, const int blo) {
	if(blo) be[x] = blo, s[blo].insert(x);
	ans += (dis[x] << 1);
	for(int i = head[x]; i; i = nxt[i]) {
		int v = dot[i];
		if(v == ba) continue;
		dis[v] = dis[x] + val[i];
		if(blo) init(v, x, blo);
		else init(v, x, ++ block);
	} 
}

void link(const int x, const int y) {
	int tmp = *s[y].begin();
	b.erase(Pair(num[be[x]], be[x])); b.erase(Pair(num[y], y));
	a.erase(Pair(tmp, y));
	res[x] = tmp;
	s[y].erase(tmp);
	if(! s[y].empty()) a.insert(Pair(*s[y].begin(), y));
	-- num[be[x]]; -- num[y];
	b.insert(Pair(num[be[x]], be[x])); b.insert(Pair(num[y], y));
}

int main() {
	int u, v, w;
	scanf("%d", &n);
	for(int i = 1; i < n; ++ i) {
		scanf("%d %d %d", &u, &v, &w);
		addEdge(u, v, w);
		addEdge(v, u, w);
	}
	maxx[0] = n;
	makeRoot(1, 0);
	init(root, 0, 0);
	s[0].insert(root);
	for(int i = 0; i <= block; ++ i) {
		a.insert(Pair(*s[i].begin(), i));
		num[i] = (int(s[i].size()) << 1);
		b.insert(Pair(num[i], i));
	}
	for(int i = 1; i <= n; ++ i) {
		if(b.rbegin() -> first == n - i + 1 && b.rbegin() -> second != be[i] && b.rbegin() -> second != 0) link(i, b.rbegin() -> second);
		else {
			set <Pair>::iterator it = a.begin();
			if(i != root && it -> second == be[i]) ++ it;
			link(i, it -> second);
		}
	}
	printf("%lld\n", ans);
	for(int i = 1; i <= n; ++ i) printf("%d%c", res[i], (i == n ? '\n' : ' '));
	return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
CodeForces - 616D是一个关于找到一个序列中最长的第k好子段的起始位置和结束位置的问题。给定一个长度为n的序列和一个整数k,需要找到一个子段,该子段中不超过k个不同的数字。题目要求输出这个序列最长的第k好子段的起始位置和终止位置。 解决这个问题的方法有两种。第一种方法是使用尺取算法,通过维护一个滑动窗口来记录\[l,r\]中不同数的个数。每次如果这个数小于k,就将r向右移动一位;如果已经大于k,则将l向右移动一位,直到个数不大于k。每次更新完r之后,判断r-l+1是否比已有答案更优来更新答案。这种方法的时间复杂度为O(n)。 第二种方法是使用枚举r和双指针的方法。通过维护一个最小的l,满足\[l,r\]最多只有k种数。使用一个map来判断数的种类。遍历序列,如果当前数字在map中不存在,则将种类数sum加一;如果sum大于k,则将l向右移动一位,直到sum不大于k。每次更新完r之后,判断i-l+1是否大于等于y-x+1来更新答案。这种方法的时间复杂度为O(n)。 以上是两种解决CodeForces - 616D问题的方法。具体的代码实现可以参考引用\[1\]和引用\[2\]中的代码。 #### 引用[.reference_title] - *1* [CodeForces 616 D. Longest k-Good Segment(尺取)](https://blog.csdn.net/V5ZSQ/article/details/50750827)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item] - *2* [Codeforces616 D. Longest k-Good Segment(双指针+map)](https://blog.csdn.net/weixin_44178736/article/details/114328999)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值