弱いペアの離散化+ツリー配列

ポータル

タイトル説明

ルートツリー上にポイントペア（u、v）がいくつあるかを調べます。これは、uがvの祖先であり、a [u] * a [v]≤kであることを満たしています。

分析

Guangguangは、この質問は議長ツリーで直接解決できると私に言いましたが、私は議長ツリーを知らないので、非常に面倒な方法でしか対処できません（次回はそれを学ぶ必要があります）

すべての重みをベクトルに格納し、重複を並べ替えて削除し、マップを使用して検索し、dfsツリーに移動します。ノードに到達するたびに、ベクトルの現在の条件を満たす最大の重みの位置を二分することができます。 、次に、ツリー配列を使用して合計し、このポイントの重みの位置に+1を追加します。これは、dfsの次のノードであり、バックトラックする場合は-1です。

コード

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <map>
#include <queue>
#include <cstring>
#define debug(x) cout<<#x<<":"<<x<<endl;
#define _CRT_SECURE_NO_WARNINGS
#pragma GCC optimize("Ofast","unroll-loops","omit-frame-pointer","inline")
#pragma GCC option("arch=native","tune=native","no-zero-upper")
#pragma GCC target("avx2")
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> PII;
const int INF = 0x3f3f3f3f;
const int N = 1e5 + 10,M = 2 * N;
int n;
ll k;
ll a[N];
vector<ll> num;
bool st[N];
int h[N],ne[M],e[M],idx;
int in[N],out[N];
int cnt;
int tr[N];
ll ans;
map<ll,int> pp;

int lowbit(int x){
    
    
    return x & -x;
}

void add(int x,int y){
    
    
    ne[idx] = h[x],e[idx] = y,h[x] = idx++;
}

int find(ll x){
    
    
    return upper_bound(num.begin(),num.end(),x) - num.begin();
}

void ad(int x,int c){
    
    
    for(int i = x;i <= n;i += lowbit(i)) tr[i] += c;
}

int sum(int x){
    
    
    int res = 0;
    for(int i = x;i;i -= lowbit(i)) res += tr[i];
    return res;
}

void dfs(int u){
    
    
    ll x = k / a[u];
    int p = find(x);
    p--;
    ans += sum(p);
    ad(pp[a[u]],1);
    for(int i = h[u];~i;i = ne[i]){
    
    
        int j = e[i];
        dfs(j);
    }
    ad(pp[a[u]],-1);
}

int main(){
    
    
    int t;
    scanf("%d",&t);
    while(t--){
    
    
        pp.clear();
        num.clear();
        idx = 0;
        scanf("%d%lld",&n,&k);
        memset(h,-1,sizeof h);
        memset(tr,0,sizeof h);
        memset(st,0,sizeof st);
        num.push_back(-1);
        for(int i = 1;i <= n;i++) {
    
    
            scanf("%lld",&a[i]);
            num.push_back(a[i]);
        }
        for(int i = 1;i < n;i++){
    
    
            int x,y;
            scanf("%d%d",&x,&y);
            add(x,y);
            st[y] = 1;
        }
        int root = -1;
        for(int i = 1;i <= n;i++){
    
    
            if(!st[i]){
    
    
                root = i;
                break;
            }
        }
        sort(num.begin(),num.end());
        num.erase(unique(num.begin(),num.end()),num.end());
        for(int i = 1;i < num.size();i++){
    
    
            pp[num[i]] = i;
        }
        ans = 0;
        dfs(root);
        printf("%lld\n",ans);
    }
}

/**
*　　┏┓　　　┏┓+ +
*　┏┛┻━━━┛┻┓ + +
*　┃　　　　　　　┃
*　┃　　　━　　　┃ ++ + + +
*  ████━████+
*  ◥██◤　◥██◤ +
*　┃　　　┻　　　┃
*　┃　　　　　　　┃ + +
*　┗━┓　　　┏━┛
*　　　┃　　　┃ + + + +Code is far away from 　
*　　　┃　　　┃ + bug with the animal protecting
*　　　┃　 　 ┗━━━┓ 神兽保佑,代码无bug　
*　　　┃ 　　　　　　 ┣┓
*　　  ┃ 　　　　　 　┏┛
*　    ┗┓┓┏━┳┓┏┛ + + + +
*　　　　┃┫┫　┃┫┫
*　　　　┗┻┛　┗┻┛+ + + +
*6
*6
*4
*/