首页 > 代码库 > BZOJ2783: [JLOI2012]树
BZOJ2783: [JLOI2012]树
题目:http://www.lydsy.com/JudgeOnline/problem.php?id=2783
题解:因为深度必须是递增的,所以就是从上到下的一条路径。
令d[i]表示i到根的路径,那么如果从x出发的满足题意的终点y必然有d[y]-d[fa[x]]=s
查询某个子树内某个值的出现次数?主席树!!!
d[i]太大?离散化就行。我省事用了map。
代码:
1 #include<cstdio> 2 3 #include<cstdlib> 4 5 #include<cmath> 6 7 #include<cstring> 8 9 #include<algorithm> 10 11 #include<iostream> 12 13 #include<vector> 14 15 #include<map> 16 17 #include<set> 18 19 #include<queue> 20 21 #include<string> 22 23 #define inf 1000000000 24 25 #define maxn 100000+5 26 27 #define maxm 2000000+5 28 29 #define eps 1e-10 30 31 #define ll long long 32 33 #define pa pair<int,int> 34 35 #define for0(i,n) for(int i=0;i<=(n);i++) 36 37 #define for1(i,n) for(int i=1;i<=(n);i++) 38 39 #define for2(i,x,y) for(int i=(x);i<=(y);i++) 40 41 #define for3(i,x,y) for(int i=(x);i>=(y);i--) 42 43 #define for4(i,x) for(int i=head[x],y=e[i].go;i;i=e[i].next,y=e[i].go) 44 45 #define for5(n,m) for(int i=1;i<=n;i++)for(int j=1;j<=m;j++) 46 47 #define mod 1000000007 48 49 #define mid ((l+r)>>1) 50 51 #define lch k<<1,l,mid 52 53 #define rch k<<1|1,mid+1,r 54 55 using namespace std; 56 57 inline int read() 58 59 { 60 61 int x=0,f=1;char ch=getchar(); 62 63 while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();} 64 65 while(ch>=‘0‘&&ch<=‘9‘){x=10*x+ch-‘0‘;ch=getchar();} 66 67 return x*f; 68 69 } 70 int n,m,ti,cnt,tot,head[maxn],a[maxn],id[maxn],pos[maxn][2],sum,fa[maxn],rt[maxn]; 71 int s[maxm],ls[maxm],rs[maxm]; 72 struct edge{int go,next;}e[maxn]; 73 map<int,int>mp; 74 inline void add(int x,int y) 75 { 76 e[++tot]=(edge){y,head[x]};head[x]=tot; 77 } 78 inline void dfs(int x) 79 { 80 pos[x][0]=++ti;id[ti]=x; 81 for4(i,x)if(!pos[y][0]) 82 { 83 fa[y]=x;a[y]+=a[x]; 84 dfs(y); 85 } 86 pos[x][1]=ti; 87 } 88 inline void update(int l,int r,int x,int &y,int z) 89 { 90 y=++sum; 91 s[y]=s[x]+1; 92 if(l==r)return; 93 ls[y]=ls[x];rs[y]=rs[x]; 94 if(z<=mid)update(l,mid,ls[x],ls[y],z);else update(mid+1,r,rs[x],rs[y],z); 95 } 96 inline int query(int l,int r,int x,int y,int z) 97 { 98 if(l==r)return s[y]-s[x]; 99 return z<=mid?query(l,mid,ls[x],ls[y],z):query(mid+1,r,rs[x],rs[y],z);100 }101 102 int main()103 104 {105 106 freopen("input.txt","r",stdin);107 108 freopen("output.txt","w",stdout);109 110 n=read();m=read();111 for1(i,n)a[i]=read();112 for1(i,n-1){int x=read(),y=read();add(x,y);}113 dfs(1);114 for1(i,n)if(!mp[a[i]])mp[a[i]]=++cnt;115 for1(i,n)update(1,cnt,rt[i-1],rt[i],mp[a[id[i]]]);116 ll ans=0;117 for1(i,n)118 {119 int t=mp[a[fa[i]]+m];120 if(t)ans+=query(1,cnt,rt[pos[i][0]-1],rt[pos[i][1]],t);121 }122 cout<<ans<<endl;123 124 return 0;125 126 }
看着status里面一堆人代码超短有种不详的预感。。。
UPD:T_T我的做法好无脑。。。
还有一种比较好想到的做法是倍增,对于一个终点倍增+二分出满足题意的起点。戳这里:http://jiruyi910387714.is-programmer.com/posts/40545.html
还有黄巨大的set做法,真是orz http://hzwer.com/4237.html
BZOJ2783: [JLOI2012]树
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。