hdu 5029 Relief grain 树链剖分

给一棵树

离线操作;

a  b  z; 在a到b路径上的点加,种类为 z 的粮食;


输出:每个点的粮食个数最多的种类数。//详见测试用例


树链剖分后,线段树为粮食的种类,在用标记数组标记下段的加值,搞搞……


#pragma comment(linker, "/STACK:200000000")
#include
#include
#include
#include
#include
#include
using namespace std;
#define ll __int64
const ll N=100010;

vector plu[100010];
vector sub[100010];
struct Edge
{
	int to,next;
}edge[N*2];
int head[N],tot;
int top[N];//top[v]表示v所在的重链的顶端节点
int fa[N]; //父亲节点
int deep[N];//深度
int num[N];//num[v]表示以v为根的子树的节点数
int p[N];//p[v]表示v与其父亲节点的连边在线段树中的位置
int fp[N];//和p数组相反
int son[N];//重儿子
int pos;
void init()
{
	tot=0;
	memset(head,-1,sizeof(head));
	pos=1;
	memset(son,-1,sizeof(son));
}
void addedge(int u,int v)
{
	edge[tot].to=v; edge[tot].next=head[u]; head[u]=tot++;
}
void dfs1(int u,int pre,int d)//第一遍dfs求出fa,deep,num,son
{
	deep[u]=d;
	fa[u]=pre;
	num[u]=1;
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v!=pre)
		{
			dfs1(v,u,d+1);
			num[u]+=num[v];
			if(son[u]==-1 || num[v]>num[son[u]])
				son[u]=v;
		}
	}
}
void getpos(int u,int sp)//第二遍dfs求出top和p
{
	top[u]=sp;
	p[u]=pos++;
	fp[p[u]]=u;
	if(son[u]==-1) return ;
	getpos(son[u],sp);
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v!=son[u] && v!=fa[u])
			getpos(v,v);
	}
}

void change(int u,int v,int val)//查询u->v边的最大值
{
	int f1=top[u],f2=top[v];
	while(f1!=f2)
	{
		if(deep[f1]deep[v]) swap(u,v);//p[u],p[v]
	plu[p[u]].push_back(val);
	sub[p[v]+1].push_back(val);
}
struct node
{
	int l,r,mx,ID;
	int mid()
	{
		return (l+r)>>1;
	}
}tree[4*N];

void build(int id,int x,int y)
{
	tree[id].l=x;
	tree[id].r=y;
	tree[id].mx=0;
	if(x==y)
	{
		tree[id].ID=x;
		return ;
	}
	int mid=(x+y)>>1;
	build(id<<1,x,mid);
	build(id<<1|1,mid+1,y);
}
void pushup(int id)
{
	if(tree[id<<1].mx>=tree[id<<1|1].mx)
	{
		tree[id].mx=tree[id<<1].mx;
		tree[id].ID=tree[id<<1].ID;
	}
	else
	{
		tree[id].mx=tree[id<<1|1].mx;
		tree[id].ID=tree[id<<1|1].ID;
	}
}
void update(int id,int x,int val)
{
	if(tree[id].l==x && tree[id].r==x)
	{
		tree[id].mx+=val;
		return ;
	}
	int mid=(tree[id].l+tree[id].r)>>1;
	if(x<=mid)
		update(id<<1,x,val);
	else
		update(id<<1|1,x,val);
	pushup(id);
}
int ans[N];
int main()
{
	int n,m,i,j,k;
	while(scanf("%d%d",&n,&m))
	{
		if(!n && !m) break;
		init();
		for(i=1;i


你可能感兴趣的:(树链剖分)