ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

树形DP·1771仓库选址

2020-01-16 14:39:16  阅读:328  来源: 互联网

标签:50000 10 1771 ba XorM 树形 星球 DP dis


在这里插入图片描述

题目

喵星系有nnn个星球,星球以及星球间的航线形成一棵树。
从星球aaa到星球bbb要花费[dis(a,b)XorM][dis(a,b)XorM][dis(a,b)XorM]秒。dis(a,b)表示a,ba,ba,b间的航线长度
求星球iii到其他所有星球花费的时间之和。
在这里插入图片描述

输入

第一行包含两个正整数nMn,Mn,M
接下来n1n−1n−1行,每行333个正整数a,b,ca,b,ca,b,c表示aba,ba,b之间的航线长度为ccc

输出

nnn行,每行一个整数,表示星球i到其他所有星球花费的时间之和。
在这里插入图片描述

样例

input
4 0
1 2 1
1 3 2
1 4 3
output
6
8
10
12

在这里插入图片描述

数据规模

测试点编号 N M
1 6 0
2 100 5
3 2000 9
4 50000 0
5 50000 0
6 50000 1
7 50000 6
8 100000 10
9 100000 13
10 100000 15

答案不超过2×10^9。
在这里插入图片描述

题解

在这里插入图片描述

代码

#include <bits/stdc++.h>
#define ll long long
using namespace std;

const int Max = 1e5+10;
struct psx{int y, v, nxt;} e[Max<<1];
int tot, n, M, hea[Max];
ll dp[Max], path[Max][16], sche[16];
//path表示i到其他点,后四位为j的路径条数
//sche后四位状态为j的向上的路径条数

inline void insert(int x, int y, int v) {
	e[++tot].nxt = hea[x];
	e[tot].y = y;
	e[tot].v = v;
	hea[x] = tot;
}

inline void init(){
	scanf("%d%d", &n, &M);
	for(int i = 1, a, b, c; i < n; i++) {
		scanf("%d%d%d", &a, &b, &c);
		insert(a, b, c);
		insert(b, a, c);
	}
}

void dfs1(int x, int fa) {
	path[x][0] = 1;
	for(int i = hea[x]; i; i = e[i].nxt) {
		int y = e[i].y, z = e[i].v;
		if(y == fa) continue;
		dfs1(y, x);
		dp[x] += dp[y];
		for(int j = 0; j < 16; j++) {
			path[x][(j+z)%16] += path[y][j];
			dp[x] += 1LL*path[y][j] * z;
		}
	}
}//子树

void dfs2(int x, int fa) {
	for(int i = hea[x]; i; i = e[i].nxt) {
		int y = e[i].y, z = e[i].v;
		if(y == fa) continue;
		int sz = 0;
		memset(sche, 0, sizeof sche);
		for(int j = 0; j < 16; j++) {
			sche[(j+z)%16] += path[x][j] - path[y][((j-z)%16+16)%16];
			sz += path[y][j];
		}
		dp[y] = dp[x] + (n - sz*2) * z;
		for(int j = 0; j < 16; j++) path[y][j] += sche[j];
		dfs2(y, x);
	}
}//换根

void calc() {//后四位
	for(int i = 1; i <= n; i++) {
		--path[i][0];//自己减自己
		for(int j = 0; j < 16; j++) {
			int val = (j ^ M) - j;
			dp[i] += 1LL * path[i][j] * val;
		}
	}
}

inline void work(){
	dfs1(1, 0);
	dfs2(1, 0);
	calc();
}

inline void outo(){
	for(int i = 1; i <= n; i++)
		printf("%d\n", dp[i]);
}

int main() {
	init();
	work();
	outo();
	return 0;
}

qyxpsx7 发布了90 篇原创文章 · 获赞 10 · 访问量 3881 私信 关注

标签:50000,10,1771,ba,XorM,树形,星球,DP,dis
来源: https://blog.csdn.net/qyxpsx7/article/details/104003550

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有