HDU 5877 Weak Pair DFS + 树状数组 + 其实不用离散化

Posted stupid_one

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了HDU 5877 Weak Pair DFS + 树状数组 + 其实不用离散化相关的知识,希望对你有一定的参考价值。

http://acm.hdu.edu.cn/listproblem.php?vol=49

给定一颗树,然后对于每一个节点,找到它的任何一个祖先u,如果num[u] * num[v] <= k。则贡献加1

思路:主要的麻烦就是动态修改前缀和了。因为对于每个数字val。则找它祖先的话, <= k / val的数字,都是合法的。所以问题转化成求你现在dfs中保存的数字,有多少个是  <= k / val的,树状数组即可。

问题就是,数字太大了,这不适合树状数组,那么我们把每个数字离散成他们的下标即可,但其实这部的操作是不需要的,就是不需要用map来离散的,因为一个数组a[],排序后的数组b[],b[i]就是a[]中某一个数字,那么i就是其对应离散后的下标,怎么找这个下标呢?lower_bound即可。但是,有相同的数字的话,lower_bound总是会找第一的,就是2 2 2 2,你找2,返回的是1的。但是没事,这不影响我们的结果。

所以,lower_bound即可,然后找到有多少个数是 <= k / val的,首先找到k / val离散后的下标,那么问题来了,没有k / val这个数值怎么办?那么就找到第一个 <= k / val的就好了。int pos = upper_bound - 1; 。这个不能用lower_bound来找下标了,为什么呢,因为它可能不存在,所以找到一个比它大的了,如果存在,那刚好,不存在又蛋疼,所以用upper_bound找,-1即可。

这题比赛的时候想到怎么做,但是写不出,然后赛后也写了很久,主要是dfs那么什么时候更新,什么时候取消更新,有点乱,还是太渣了啊。。。努力吧

我dfs的思路是这样的:首先没到一个节点cur。算贡献,然后这个节点入树状数组,然后当这个节点的所有边遍历完后,把这个数删除,注意一点是数字可能有0,不能除。

技术分享
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
#define inf (0x3f3f3f3f)
typedef long long int LL;

#include <iostream>
#include <sstream>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <string>
const int maxn = 1e5 + 20;
LL a[maxn];
LL b[maxn];
int n;
LL k;
struct edge {
    int u, v;
    int tonext;
} e[maxn];
int num, first[maxn];
bool in[maxn];
void addedge(int u, int v) {
    ++num;
    e[num].u = u;
    e[num].v = v;
    e[num].tonext = first[u];
    first[u] = num;
}
LL ans;
int c[maxn];//
int lowbit (int x) { //
    return x&(-x);
}
void add (int pos,int val) { //
    if (pos <= 0) return;
    while (pos<=n) { //
        c[pos] += val;
        pos += lowbit(pos);
    }
    return ;
}
int get_sum (int pos) { //???1--pos???
    if (pos <= 0) return 0;
    int ans = 0;
    while (pos) {
        ans += c[pos];
        pos -= lowbit(pos);
    }
    return ans;
}
void calc(int cur) {
    if (a[cur] == 0) {
        ans += get_sum(n);
    } else {
        LL tofind = k / a[cur];
        int pos = upper_bound(b + 1, b + 1 + n, tofind) - b;
        ans += get_sum(pos - 1);
        //cout << tofind << "   *****" << pos - 1 << endl;
    }
//    cout << cur << " " << ans << endl;
}
void dfs(int cur) {
    calc(cur);
    int pos = lower_bound(b + 1, b + 1 + n, a[cur]) - b;
    add(pos, 1);
    for (int i = first[cur]; i; i = e[i].tonext) {
        dfs(e[i].v);
    }
    add(pos, -1);
}
void work() {
    scanf("%d%I64d", &n, &k);
    for (int i = 1; i <= n; ++i) {
        scanf("%I64d", &a[i]);
        b[i] = a[i];
    }
//    for (int i = 1; i <= n; ++i) {
//        cout << a[i] << " ";
//    }
//    cout << endl;
    sort(b + 1, b + 1 + n);
//    for (int i = 1; i <= n; ++i) {
//        cout << b[i] << " ";
//    }
//    cout << endl;
    for (int i = 1; i <= n - 1; ++i) {
        int u, v;
        scanf("%d%d", &u, &v);
        addedge(u, v);
        in[v] = 1;
    }
    int root = -inf;
    for (int i = 1; i <= n; ++i) {
        if (in[i] == 0) {
            root = i;
            break;
        }
    }
    ans = 0;
    dfs(root);
//    cout << ans << endl;
    printf("%I64d\n", ans);
    return ;
}
int main () {
#ifdef LOCAL
    freopen("data.txt","r",stdin);
#endif
    int t;
    scanf("%d", &t);
    while (t--) {
        work();
        num = 0;
        memset(in, 0, sizeof in);
        memset(c, 0, sizeof c);
        memset(first, 0, sizeof first);
    }
    return 0;
}
View Code

 

以上是关于HDU 5877 Weak Pair DFS + 树状数组 + 其实不用离散化的主要内容,如果未能解决你的问题,请参考以下文章

HDU - 5877 Weak Pair (dfs+树状数组)

HDU 5877 Weak Pair(树状数组+dfs+离散化)

HDU 5877 Weak Pair DFS + 树状数组 + 其实不用离散化

HDU5877 - Weak Pair

2016 ACM/ICPC Asia Regional Dalian Online HDU 5877 Weak Pair treap + dfs序

HDU 5877 Weak Pair