HDU 5877 Weak Pair
2017-08-27 15:15
267 查看
题意:问一棵树中有多少序数对 < u , v >满足,u是v的祖先节点,并且这两个节点的值相乘小于或等于k。
分析:感觉这个思路还是很妙的。考虑深搜的过程,当搜到一个节点的时候,所有经过的点都是它的祖先节点。所以就可以每搜到一个点就计算经过的点中有多少个点满足和它相乘<=k。这一步可以用树状数组来计算。然后再把这个点加入树状数组。
分析:感觉这个思路还是很妙的。考虑深搜的过程,当搜到一个节点的时候,所有经过的点都是它的祖先节点。所以就可以每搜到一个点就计算经过的点中有多少个点满足和它相乘<=k。这一步可以用树状数组来计算。然后再把这个点加入树状数组。
#include<cstdio> #include<algorithm> #include<queue> #include<vector> #include<cstring> #include<iostream> using namespace std; const int N = 2e5 + 5; typedef long long ll; ll a , s , ans, n, m; int f , root, t; vector<int> son ; struct p{ ll k, id, sid; }b ; int cmp1(p a, p b){ return a.k < b.k; } int cmp2(p a, p b){ return a.id < b.id; } int getId(ll p){ int l = 0, r = n, ans = 0; while(l <= r){ int mid = (l + r) / 2; if(b[mid].k <= p){ ans = mid; l = mid + 1; }else r = mid - 1; } return ans; } void set(int k, int t){ while(k <= n){ s[k] += t; k += k & (-k); } } ll query(int k){ ll ans = 0; while(k > 0){ ans += s[k]; k -= k & (-k); } return ans; } void dfs(int k){ // cout << k << endl; t = b[a[k]].k == 0 ? 0 : getId(m / b[a[k]].k); ans += query(t); if(son[k].size() == 0){ return; } set(a[k], 1); for(int i = 0; i < son[k].size(); i++){ dfs(son[k][i]); } set(a[k], -1); } int main(){ int T; //freopen("c.txt", "r", stdin); scanf("%d", &T); for(; T--; ){ ans = 0; memset(s, 0, sizeof(s)); memset(f, 0, sizeof(f)); scanf("%lld%lld", &n, &m); for(int i = 1; i <= n; i++){ son[i].clear(); scanf("%lld", &b[i].k); b[i].id = i; } sort(b + 1, b + n + 1, cmp1); for(int i = 1; i <= n; i++){ a[b[i].id] = i; } for(int i = 1; i < n; i++){ int u, v; scanf("%d%d", &u, &v); f[v]++; son[u].push_back(v); } for(int i = 1; i <= n; i++) if(f[i] == 0) dfs(i);; printf("%lld\n", ans); } return 0; }
相关文章推荐
- HDU 5877 Weak Pair (2016年大连网络赛 J dfs+反向思维)
- HDU 5877 Weak Pair 【dfs】【树状数组】【离散化】
- hdu_5877_Weak Pair(离散+DFS+树状数组)
- HDU 5877 Weak Pair
- HDU - 5877 Weak Pair (dfs+离散化+树状数组)
- 2016 ICPC 大连网络赛 HDU 5877 Weak Pair
- HDU 5877 Weak Pair
- HDU 5877 Weak Pair(dfs+BIT)
- hdu 5877 Weak Pair
- HDU 5877 Weak Pair(2016 ACM/ICPC Asia Regional Dalian Online)
- HDU 5877 Weak Pair(dfs + 树状数组 + 离散化)
- HDU 5877 2016大连网络赛 Weak Pair(树状数组,线段树,动态开点,启发式合并,可持久化线段树)
- HDU 5877 - Weak Pair
- HDU 5877 2016 ACM/ICPC Asia Regional Dalian Online 1010 Weak Pair
- HDU 5877 2016大连网络赛 Weak Pair(树状数组,线段树,动态开点,启发式合并,可持久化线段树)
- HDU 5877 Weak Pair 树状数组 + DFS
- hdu 5877 Weak Pair(树状数组)
- HDU 5877 Weak Pair(离散化+dfs+树状数组)——2016 ACM/ICPC Asia Regional Dalian Online
- hdu 5877 - Weak Pair (2016大连网络赛) 离散化 + 树状数组
- HDU 5877 Weak Pair(离散化+dfs+树状数组) 大连区域网络赛