关闭

HYSBZ 1036 树的统计Count(树链剖分)

150人阅读 评论(0) 收藏 举报
分类:

思路:树链剖分的又一个模板题,不过坑了好久...不知道为什么输入询问的时候用string会RE...


#include<iostream>
#include<cmath>
#include<cstdio>
#include<vector>
#include<cstring>
#include<string>
using namespace std;
const int maxn = 30005;
#define lson i<<1,l,m
#define rson i<<1|1,m+1,r
int n,tot;
int dep[maxn],son[maxn],siz[maxn],fa[maxn],top[maxn],id[maxn];
vector<int>e[maxn];
int val[maxn];
int maxv[maxn*4];
int sum[maxn*4];
int temp[maxn];
void push_up(int i)
{
	maxv[i] = max(maxv[i<<1],maxv[i<<1|1]);
	sum[i] = sum[i<<1]+sum[i<<1|1];
}
void build(int i,int l,int r)
{
	maxv[i]=-0x3f3f3f3f;
	if(l==r)
	{
		maxv[i]=val[l];
		sum[i]=val[l];
		return;
	}
	int m = (l+r)>>1;
	build(lson);
	build(rson);
	push_up(i);
}
void update(int v,int vall,int i,int l,int r)
{

    if(l==r)
	{
		sum[i] = vall;
		maxv[i]=vall;
		return;
	}
	int m = (l+r)>>1;
	if(v <= m)
		update(v,vall,lson);
	else
		update(v,vall,rson);
	push_up(i);
}
int query1(int ql,int qr,int i,int l,int r)
{
	if(ql<=l && qr>=r)
		return sum[i];
	int m = (l+r)>>1;
	int ans = 0;
	if(ql<=m)
		ans+=query1(ql,qr,lson);
	if(qr>m)
		ans+=query1(ql,qr,rson);
	return ans;
}
int query2(int ql,int qr,int i,int l,int r)
{
	if(ql<=l && qr>=r)
		return maxv[i];
	int m = (l+r)>>1;
	int ans = -1e9;
	if(ql<=m)
		ans = max(query2(ql,qr,lson),ans);
	if(qr>m)
		ans=max(query2(ql,qr,rson),ans);
	return ans;

}
void dfs1(int u,int f,int d)
{
	dep[u]=d;
	fa[u]=f;
	son[u]=0;
	siz[u]=1;
	for(int i = 0;i<e[u].size();i++)
	{
		int v = e[u][i];
		if(v==f)
			continue;
		dfs1(v,u,d+1);
		siz[u]+=siz[v];
		if (siz[son[u]] < siz[v])
			son[u]=v;
	}
}
void dfs2(int u,int tp)
{
	top[u]=tp;
	id[u]=++tot;
	if(son[u])
		dfs2(son[u],tp);
	for(int i = 0;i<e[u].size();i++)
	{
		int v = e[u][i];
		if(v==fa[u]|| v==son[u])
			continue;
		dfs2(v,v);
	}
}

int Yougth2(int u,int v)
{
	int tp1 = top[u],tp2 = top[v];
	int ans = -1e9;
	while(tp1!=tp2)
	{
		if(dep[tp1] < dep[tp2])
		{
			swap(tp1,tp2);
			swap(u,v);
		}
		ans = max(query2(id[tp1],id[u],1,1,tot),ans);
		u = fa[tp1];
		tp1 = top[u];
	}
	if(dep[u] > dep[v])
		swap(u,v);
	ans=max(query2(id[u],id[v],1,1,tot),ans);
	return ans;
}

int Yougth1(int u,int v)
{
	int tp1 = top[u],tp2 = top[v];
	int ans = 0;
	while(tp1!=tp2)
	{
		if(dep[tp1] < dep[tp2])
		{
			swap(tp1,tp2);
			swap(u,v);
		}
		ans+=query1(id[tp1],id[u],1,1,tot);
		u = fa[tp1];
		tp1 = top[u];
	}
	if(dep[u] > dep[v])
		swap(u,v);
	ans+=query1(id[u],id[v],1,1,tot);
	return ans;
}
int main()
{
    while(scanf("%d",&n)!=EOF)
	{
		tot = 0;
		memset(val,0,sizeof(val));
		memset(son,0,sizeof(son));
		for(int i = 0;i<=n;i++)
			e[i].clear();
		for(int i =1;i<n;i++)
		{
			int u,v;
			scanf("%d%d",&u,&v);
			e[u].push_back(v);
			e[v].push_back(u);
		}
		for(int i = 1;i<=n;i++)
			scanf("%d",&temp[i]);
		dfs1(1,0,1);
		dfs2(1,1);
		for(int i = 1;i<=n;i++)
			val[id[i]]=temp[i];
		build(1,1,tot);
		int q;
		scanf("%d",&q);
		while(q--)
		{
		    char s[105];
			int a,b;
			scanf("%s%d%d",s,&a,&b);
			if(s[1] == 'M')
			{
                  printf("%d\n",Yougth2(a,b));
			}
			if(s[1]=='S')
			{
                  printf("%d\n",Yougth1(a,b));
			}
			if (s[1]=='H')
			{
                 update(id[a],b,1,1,tot);
			}
		}
	}
}



Description

  一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成
一些操作: I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I
II. QSUM u v: 询问从点u到点v的路径上的节点的权值和 注意:从点u到点v的路径上的节点包括u和v本身

Input

  输入的第一行为一个整数n,表示节点的个数。接下来n – 1行,每行2个整数a和b,表示节点a和节点b之间有
一条边相连。接下来n行,每行一个整数,第i行的整数wi表示节点i的权值。接下来1行,为一个整数q,表示操作
的总数。接下来q行,每行一个操作,以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。 
对于100%的数据,保证1<=n<=30000,0<=q<=200000;中途操作中保证每个节点的权值w在-30000到30000之间。

Output

  对于每个“QMAX”或者“QSUM”的操作,每行输出一个整数表示要求输出的结果。

Sample Input

4
1 2
2 3
4 1
4 2 1 3
12
QMAX 3 4
QMAX 3 3
QMAX 3 2
QMAX 2 3
QSUM 3 4
QSUM 2 1
CHANGE 1 5
QMAX 3 4
CHANGE 3 6
QMAX 3 4
QMAX 2 4
QSUM 3 4

Sample Output

4
1
2
2
10
6
5
6
5
16



0
0
查看评论

HYSBZ 1036 树的统计Count(树链剖分)

HYSBZ 1036 树的统计Count 题目链接 就树链剖分,线段树维护sum和maxx即可 代码: #include #include #include #include using namespace std; const int N = 30005; in...
  • u011217342
  • u011217342
  • 2014-10-01 00:48
  • 1235

树链剖分教程 & bzoj 1036 [ZJOI2008] 树的统计 Count 题解

【原题】 1036: [ZJOI2008]树的统计Count Time Limit: 10 Sec  Memory Limit: 162 MB Submit: 4465  Solved: 1858 [Submit][...
  • u013724185
  • u013724185
  • 2014-04-28 22:27
  • 5702

HYSBZ 1036 树的统计Count(树链剖分)

简单题,1A #define _CRT_SECURE_NO_WARNINGS #include #include #include #include using namespace std; #define MAX 30030 #define ls rt<<1 #define rs l...
  • u013611908
  • u013611908
  • 2015-04-02 10:54
  • 313

hysbz 1036 树的统计Count(树链剖分)

题目链接:hysbz 1036 树的统计Count 题目大意:略。 解题思路:树链剖分+线段树维护。 #include #include #include using namespace std; const int maxn = 30005; const int INF =...
  • u011328934
  • u011328934
  • 2014-10-17 09:39
  • 571

HYSBZ - 1036 树的统计Count (树链剖分)

树链剖分就是将一个树的路径转化成重链和轻链,对其节点或者边就行编号,从而可以转化成其他的数据结构问题来解决问题。树链剖分有些类似莫队,也就是相当于把树的路径进行了分块。数组的含义:fa[]每个节点的父节点 num[]每个节点子节点的个数(包含自己) son[v]与v在同一重链的重儿子 top[...
  • blue_kid
  • blue_kid
  • 2017-12-12 14:04
  • 47

hysbz 1036 树的统计Count 树链剖分

板题。 #include #include #include #include #define maxn 30005 #define inf 0x3fffffff using namespace std; int dep[maxn],top[maxn],fa[maxn],son[maxn],si...
  • u013665921
  • u013665921
  • 2014-11-14 23:03
  • 333

HYSBZ 1036 树的统计Count 树链剖分

题目:http://www.lydsy.com/JudgeOnline/problem.php?id=1036 题意:Description   一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成 一些操作: I. CHANGE...
  • discreeter
  • discreeter
  • 2016-05-14 20:27
  • 250

HYSBZ 1036 树的统计Count 树链剖分

这题就是裸的树链剖分传送门:http://www.lydsy.com/JudgeOnline/problem.php?id=1036#include <map> #include <set> #include <queue> #include <cmath&...
  • NMfloat
  • NMfloat
  • 2015-09-29 17:56
  • 186

HYSBZ - 1036 树的统计Count(树链剖分)

HYSBZ - 1036 树的统计Count(树链剖分) Description   一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成 一些操作: I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v...
  • cherish0222
  • cherish0222
  • 2017-08-29 09:19
  • 120

HYSBZ - 1036 树的统计Count(树剖)

题目大意:中文题解题思路:树剖裸题#include <cstdio> #include <cstring> #include <algorithm> using namespace std; const int N = 30010; const int M = 3...
  • L123012013048
  • L123012013048
  • 2015-10-13 23:33
  • 228
    个人资料
    • 访问:345333次
    • 积分:13781
    • 等级:
    • 排名:第1100名
    • 原创:1084篇
    • 转载:3篇
    • 译文:0篇
    • 评论:53条
    最新评论