题意:给出一颗标号为1-n的树,求每个节点的子树节点的标号有多少小于该节点的标号。
此题DFS序+主席树过不了,内存太小。。。
看到子树很容易想到树的DFS序,把树转一下DFS序就是求n个区间,每个区间的数字小于k的数目的问题了。
显然可以用主席树,但是内存还是太蛋疼。。。 如果发现这个序列是一个全排列,且每次询问的都是不一样的k。
那么我们可以先把所以节点的贡献置1,,询问n的子树,再把n节点的贡献置0,再询问n-1的子树。。。
这就是一个很简单的树状数组问题了。。
# include <cstdio> # include <cstring> # include <cstdlib> # include <iostream> # include <vector> # include <queue> # include <stack> # include <map> # include <set> # include <cmath> # include <algorithm> using namespace std; # define lowbit(x) ((x)&(-x)) # define pi 3.1415926535 # define eps 1e-9 # define MOD 1000000009 # define INF 1000000000 # define mem(a,b) memset(a,b,sizeof(a)) # define FOR(i,a,n) for(int i=a; i<=n; ++i) # define FO(i,a,n) for(int i=a; i<n; ++i) # define bug puts("H"); # define lch p<<1,l,mid # define rch p<<1|1,mid+1,r # define mp make_pair # define pb push_back typedef pair<int,int> PII; typedef vector<int> VI; # pragma comment(linker, "/STACK:1024000000,1024000000") typedef long long LL; int Scan() { int res=0, flag=0; char ch; if((ch=getchar())=='-') flag=1; else if(ch>='0'&&ch<='9') res=ch-'0'; while((ch=getchar())>='0'&&ch<='9') res=res*10+(ch-'0'); return flag?-res:res; } void Out(int a) { if(a<0) {putchar('-'); a=-a;} if(a>=10) Out(a/10); putchar(a%10+'0'); } const int N=100005; //Code begin... struct Edge{int p, next;}edge[N<<1]; struct Node{int l, r;}node[N]; int head[N], cnt=1, pos, ans[N], n; int tree[N]; void add_edge(int u, int v){ edge[cnt].p=v; edge[cnt].next=head[u]; head[u]=cnt++; } void dfs(int x, int fa){ node[x].l=++pos; for (int i=head[x]; i; i=edge[i].next) { int v=edge[i].p; if (v==fa) continue; dfs(v,x); } node[x].r=pos; } void add(int x, int val){while (x<=n) tree[x]+=val, x+=lowbit(x);} int query(int x){ int res=0; while (x) res+=tree[x], x-=lowbit(x); return res; } int main () { int p, u, v; while (scanf("%d%d",&n,&p), n+p) { mem(head,0); cnt=1; pos=0; mem(tree,0); FO(i,1,n) scanf("%d%d",&u,&v), add_edge(u,v), add_edge(v,u); dfs(p,0); FOR(i,1,n) add(i,1); for (int i=n; i>=1; --i) { ans[i]=query(node[i].r)-query(node[i].l); add(node[i].l,-1); } FOR(i,1,n) printf(i==1?"%d":" %d",ans[i]); putchar(' '); } return 0; }