标签:io os for sp 问题 on c log amp
题目大意:
给出一棵树。
然后有m个操作,每个操作都在两点的路径上分配不同的粮食。
最后要求输出所有村庄有的最多的粮食的种类。
思路分析:
一眼就看得出来是树链剖分的题目。
现在的问题就是,每一次操作,如何维护每个点的最多的食物编号,以及最多的食物的数量。要记录这两个值是肯定的。
首先可以想到将所有的操作按照z排序。这样每一回合操作,称一回合为处理同一种颜色。一回合结束之后,后面的操作就不会影响前面取得的最大值。因为后面的操作的食物的种类不一样,是不可以再继续累加的。
然后可以发现一个规律,就是每处理完一种颜色之后,在线段树上,颜色比这种颜色编号大的颜色都存在了这个节点的树形结构的上面。也就是父亲节点甚至是父亲节点的父亲节点。
所以我们就直接考虑pushdown部分吧,这是这道题目的难点。
假设操作只有 2 3 4这三种食物。
我们首先先处理完了所有食物种类为2的所有操作。
现在在处理种类为3的操作。如果这个操作的指定区间的lazy标记的颜色也是3.那么就直接累加,毫无疑问。但是如果这个区间是2...由于我们现在还在处理3,你不确定后面还有多少种类为3的操作,所以我们就要将这个种类为2的lazy往下推(用递归的pushdown实现)。然后将3更新。
按照如上操作处理了3之后。
现在我们处理种类为4的。我们注意到之前说的规律。
2一定在3下面,那么4的下面也之后有3.
所以现在种类4的操作,碰到了4.那么也是直接累加无疑问。好,如果碰到操作三,按照我们上述的,要将3pushdown。但是,在pushdown3的过程中,这个3又会遇到lazy颜色为2的。产生冲突。但是我们注意到,如果是4将3推下去,而且3又碰到了2,也就以为着,这个子树只有一个节点的lazy是3了。为什么,还是那个规律。自己想一下就明白。
所以意思就是下面不会有三,那个节点的全部三就是所有操作的全部三。当递推下的3的数量比碰到的那个2的数量还要小,也就意味着以下的节点的答案不可能是3 的。
但是如果2比三小,可能这下面还有一些2没有累加,所以就把2递推下去,再递推3下去,再更新4.
好,现在我们分析时间复杂度。
首先m此操作。
然后查找熟练logn
线段树更新logn
那么这个pushdown的递归呢。
因为递归层数是不会超过logn的。
所以最多也就是 m*log^3...
当然最优解肯定不是这个。。。
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <utility>
#pragma comment(linker,"/STACk:10240000,10240000")
#define maxn 100005
#define lson num<<1,s,mid
#define rson num<<1|1,mid+1,e
#define mid ((s+e)>>1)
using namespace std;
int next[maxn<<1],to[maxn<<1],head[maxn],tot;//临界表
int pre[maxn],root[maxn],siz[maxn],son[maxn],w[maxn],dep[maxn],id;//原树的父亲 链上的根 siz 有最大siz的子树 重新分配的id 深度 getid中来计数的
pair<int,int>ans[maxn<<2];//线段树变量
pair<int,int>cov[maxn<<2];
int n,cur;
void init()
{
pre[1]=0;
dep[1]=0;
tot=0;id=0;
memset(head,0,sizeof head);
}
void addedge(int u,int v)
{
tot++;
next[tot]=head[u];
to[tot]=v;
head[u]=tot;
}
void dfs(int now)//to get size,son,dep,pre...
{
son[now]=0;
siz[now]=1;
for(int p =head[now]; p ; p=next[p])
{
int t=to[p];
if(t!=pre[now])
{
pre[t]=now;
dep[t]=dep[now]+1;
dfs(t);
if(siz[t]>siz[son[now]])son[now]=t;
siz[now]+=siz[t];
}
}
}
void getid(int now,int rt)//to get w and root...
{
w[now]=++id;
root[now]=rt;
if(son[now])getid(son[now],rt);
for(int p = head[now] ; p ; p=next[p])
{
int t=to[p];
if(t!=son[now]&&t!=pre[now])
getid(t,t);
}
}
void pushdown(int num,int s,int e)
{
if(s==e)
{
cov[num].first=cov[num].second=0;
return;
}
if(cov[num].second)
{
if(cov[num<<1].second==cov[num].second)
cov[num<<1].first+=cov[num].first;
else
{
if(cov[num<<1].first<cov[num].first || cov[num].second==cur)
{
pushdown(lson);
cov[num<<1]=cov[num];
}
}
if(s==mid)
{
if(cov[num<<1].first>ans[num<<1].first)
ans[num<<1]=cov[num<<1];
}
if(cov[num<<1|1].second==cov[num].second)
cov[num<<1|1].first+=cov[num].first;
else
{
if(cov[num<<1|1].first<cov[num].first || cov[num].second==cur)
{
pushdown(rson);
cov[num<<1|1]=cov[num];
}
}
if(mid+1==e)
{
if(cov[num<<1|1].first>ans[num<<1|1].first)
ans[num<<1|1]=cov[num<<1|1];
}
cov[num].first=cov[num].second=0;
}
}
void build(int num,int s,int e)
{
ans[num].first=ans[num].second=0;
cov[num].first=0;
cov[num].second=0;
if(s==e)return;
build(lson);
build(rson);
}
void update(int num,int s,int e,int l,int r,int val)
{
if(l<=s && r>=e)
{
if(cov[num].second==val)
cov[num].first++;
else
{
pushdown(num,s,e);
cov[num].first++;
cov[num].second=val;
}
if(s==e){
if(cov[num].first>ans[num].first)
ans[num]=cov[num];
}
return;
}
pushdown(num,s,e);
if(l<=mid)update(lson,l,r,val);
if(r>mid)update(rson,l,r,val);
}
int query(int num,int s,int e,int pos)
{
if(s==e)return ans[num].second;
pushdown(num,s,e);
if(pos<=mid)return query(lson,pos);
else return query(rson,pos);
}
void work(int x,int y,int val)
{
while(root[x]!=root[y])
{
if(dep[root[x]]<dep[root[y]])swap(x,y);
update(1,1,n,w[root[x]],w[x],val);
x=pre[root[x]];
}
if(dep[x]>dep[y])swap(x,y);
update(1,1,n,w[x],w[y],val);
}
struct node
{
int x,y,z;
bool operator < (const node &cmp)const{
return z<cmp.z;
}
}line[maxn];
int main()
{
int m;
while(scanf("%d%d",&n,&m)!=EOF)
{
if(n==0 && m==0)break;
init();
for(int i=1;i<=n-1;i++)
{
int s,e;
scanf("%d%d",&s,&e);
addedge(s,e);
addedge(e,s);
}
dfs(1);
getid(1,1);
build(1,1,n);
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&line[i].x,&line[i].y,&line[i].z);
}
sort(line+1,line+1+m);
for(int i=1;i<=m;i++){
int x,y,z;
cur=line[i].z;
work(line[i].x,line[i].y,line[i].z);
}
cur=-1;
for(int i=1;i<=n;i++)
{
printf("%d\n",query(1,1,n,w[i]));
}
}
return 0;
}
/*
2 4
1 2
1 1 1
1 2 2
2 2 2
2 2 1
5 3
1 2
3 1
3 4
5 3
2 3 3
1 5 2
3 3 3
5 4
1 2
2 3
3 4
4 5
1 5 1
1 3 2
1 2 2
1 1 3
5 5
1 2
2 3
3 4
4 5
1 1 1
1 2 1
1 3 2
1 5 2
1 3 2
5 5
1 2
2 3
3 4
4 5
1 2 1
1 2 1
1 3 2
1 1 2
1 1 2
5 10
1 2
2 3
3 4
4 5
1 1 1
1 1 1
1 1 2
1 1 2
1 1 1
1 1 2
1 1 3
1 1 3
1 1 3
1 1 3
0 0
*/
标签:io os for sp 问题 on c log amp
原文地址:http://blog.csdn.net/u010709592/article/details/39482639