Triples of Cows

news/2024/7/15 17:39:26 标签: 图论, 算法

题目传送门

模拟赛 T 4 T4 T4 , 变换挺妙的, 而且感觉转换后问题就迎刃而解了

解法

强行模拟拆点重连边显然不行,会让图的边数达到 n 2 n^2 n2 级别的
——————————————————————————————————————————————————
考虑转化:在每一条边上新建一个虚点,
例如 第 i i i 条边 u − v u-v uv ,我们建一个虚点 n + i n+i n+i 并将原边变为两条边 u   −   n + i , n + i   −   v u\ -\ n+i,n+i \ - \ v u  n+i,n+i  v
发现转化后,对于删除 i i i 点的操作,只需将 i i i 删除并且合并虚点即可 ,图的形态仍是一颗树,就比较好处理了
——————————————————————————————————————————————————
考虑答案的计算 :需要分类讨论一下
首先我们记 原图编号 ≤ n \le n n 的点为实点(与虚点相对应), f u f_u fu u u u 的一级儿子的个数, g u g_u gu u u u 的二级儿子的个数, h u h_u hu u u u 的三级儿子的个数
对答案的贡献的情形分三种,记贡献为 Δ \Delta Δ
1.

在这里插入图片描述
2.
在这里插入图片描述
3.
在这里插入图片描述

将三种贡献加起来就有:
A n s = ∑ i = 1 n g i 2 − ∑ i = n + 1 n ∗ 2 + 1 f i ( f i − 1 ) ( f i + 1 ) − f i 2 + 2 f i h i Ans=\sum_{i=1}^{n} g_i^2 -\sum_{i=n+1}^{n*2+1} f_i (f_{i}-1)(f_i+1)-f_i^2+2f_ih_i Ans=i=1ngi2i=n+1n2+1fi(fi1)(fi+1)fi2+2fihi
合并虚点时用并查集维护并更新 f , g , h f,g,h f,g,h 的值

#include <algorithm>
#include <iostream>
#include <cstring>
#include <vector>

using ll = long long;
using namespace std;
const int N = 5e5 + 7;

int n;
vector<int> G[N];
ll f[N],g[N],h[N],fa[N],a[N];

int find(int x) {  return x==a[x]?x:a[x]=find(a[x]);  }

void dfs(int u,int ff) {
	fa[u]=ff;
	for(int v:G[u]) if(v!=ff) {
		dfs(v,u);
		if(u<=n) {
			g[u] += f[v] ;
		}else {
			++f[u] ;
			h[u] += g[v] ;
		}
	}
}

void work() {
	ll ans=0;
	for(int i=1;i<=n;i++) 
		ans += g[i]*g[i] ;
	for(int i=n+1;i<=(n<<1)-1;i++) 
		ans+=(f[i]*(f[i]+1)*(f[i]-1) - f[i]*f[i] + 2*f[i]*h[i]) , a[i] = i ;
	
	printf("%lld\n" , ans) ;
	for(int u=1;u<n;u++) {
		int x,y,z; x=find(fa[u]); y=fa[x]; z=find(fa[y]); // 最多影响往上3代
		ans-= g[u]*g[u] ;
		ans-=(f[x]*(f[x]+1)*(f[x]-1) - f[x]*f[x] + 2*f[x]*h[x]) ;
		ans-= g[y]*g[y];
		if(z) ans-=(f[z]*(f[z]+1)*(f[z]-1) - f[z]*f[z] + 2*f[z]*h[z]) ;
		--f[x],--g[y] ;
		if(z) --h[z];
		for(int v:G[u]) if(v!=fa[u]) {
			a[v] = x;
			f[x] += f[v]; h[x] -= f[v]; //三代变一代
			g[y] += f[v];
			h[x] += h[v];
			if(z) h[z]+=f[v] ;
			ans-=(f[v]*(f[v]+1)*(f[v]-1) - f[v]*f[v] + 2*f[v]*h[v]) ;
		}
		ans+= (f[x]*(f[x]+1)*(f[x]-1) - f[x]*f[x] + 2*f[x]*h[x]) ;
		ans+= g[y]*g[y];
		if(z) ans+=(f[z]*(f[z]+1)*(f[z]-1) - f[z]*f[z] + 2*f[z]*h[z]) ;
		
		printf("%lld\n" , ans) ;
	}
}

int main() {
    scanf("%d",&n);
    for(int i=1;i<n;i++) {
    	int u,v; scanf("%d%d",&u,&v);
    	G[u].push_back(n+i); G[n+i].push_back(v);
    	G[v].push_back(n+i); G[n+i].push_back(u);
	}
	dfs(n,0);
	work();
}

额,代码实现还是需要一点经验的,小看它了


http://www.niftyadmin.cn/n/5165963.html

相关文章

node插件MongoDB(三)—— 库mongoose 的使用

前言 提示&#xff1a;使用mongoose 的前提是你安装了node和 MongoDB。 mongoose 官网文档&#xff1a;http://mongoosejs.net/docs/index.html 文章目录 前言一、安装二、基本使用1. 打开bin目录的mongod.exe文件2. 基本使用的代码&#xff08;连接mongodb 服务&#xff09;3.…

Leetcode—102.二叉树的层序遍历【中等】

2023每日刷题&#xff08;二十四&#xff09; Leetcode—102.二叉树的层序遍历 C语言BFS实现代码 /*** Definition for a binary tree node.* struct TreeNode {* int val;* struct TreeNode *left;* struct TreeNode *right;* };*/ /*** Return an array of arr…

基于 Python 的课程助教智能聊天机器人

温馨提示&#xff1a;文末有 CSDN 平台官方提供的学长 Wechat / QQ 名片 :) 1. 项目简介 课程助教是高校中一种常见的教学模式,其在学生理论知识的掌握与实践能力的提高方面起到关键性的作用,已经成为高校日常教育环节中不可或缺的一环。然而,传统的人力助教有若干关键问题亟待…

17.复制字符串 ,包括\0

#include<stdio.h> #include <cstring>int main(){int len1,len2;char s1[44];char s2[33];scanf("%s",s1);scanf("%s",s2);len1strlen(s1)1;printf("先s1的字符长度为&#xff1a;%d\n",len1) ;strcpy(s1,s2) ;printf("复制字…

Go相关命令说明

目录 Go相关命令说明go mod tidy &#xff1a;清理未使用依赖项&#xff0c;并更新模块文件主要功能好处 go clean -modcache &#xff1a;清除模块缓存go clean -testcache &#xff1a;清除测试缓存go test -v ./client &#xff1a;测试当前目录下client目录中的所有测试函数…

Flink SQL -- 概述

1、Flink SQL中的动态表和连续查询 1、动态表&#xff1a; 因为Flink是可以做实时的&#xff0c;数据是在不断的变化的&#xff0c;所以动态表指的是Flink中一张实时变换的表&#xff0c;表中会不断的有新的数据。但是这张表并不是真正的物理表。 2、连续查询&#xff1a; 连续…

SPASS-描述性分析

将身高移入变量 结果展示&#xff1a; 表中分析变量“身高”的个案数、所有个案中的极大值、极小值、均值、标准差及偏度和峰度

Java数据的基本(原始)类型和引用类型的特点差别

本文作为“Java数据类型”一文的补充https://blog.csdn.net/cnds123/article/details/110517272 Java的数据类型可以分为基本类型&#xff08;primitive types&#xff09;和引用类型&#xff08;reference types&#xff09;两大类。在实际编程中&#xff0c;要根据需求选择合…