Почему TLE после инкапсуляции двух векторных операций в одну встроенную / макро функцию? - PullRequest
Новая
       30

Почему TLE после инкапсуляции двух векторных операций в одну встроенную / макро функцию?

0 голосов
/ 27 апреля 2020

Я использую «вектор G [N]», чтобы сохранить недирективный граф (N = 2e5 + 10), как вы знаете, для каждого ребра потребуются следующие операции (a, b): 

G[a].push_back(b); G[b].push_back(a);

Это A C, если использовать их напрямую, но после попытки инкапсулировать их в одну встроенную функцию или макрос, это будет TLE. так почему? Насколько я понимаю, встроенная функция или макрос точно так же, как две операции (потому что они заменят место функции / макроса с ними во время предварительной компиляции).

Весь код только для вашей справки, кстати это решение для atcoder abs-163-F . 

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int N = 2e5 + 10;
int n, c[N], cnt[N], siz[N];
inline LL cal(int n) {
    // C(n,2) + n
    return 1LL*n*(n+1)/2;
}
LL ans[N];
vector<int>G[N];
void dfs(int v, int fa) {
    int tmp = cnt[c[v]];
    siz[v] = 1;
    for(int u:G[v]) {
        if(u==fa) continue;
        cnt[c[v]] = 0;
        dfs(u, v);
        siz[v] += siz[u];
        ans[c[v]]+=cal(siz[u]-cnt[c[v]]);
    }
    cnt[c[v]] = tmp + siz[v];
}

#define ADD(a, b)   \
    do {            \
        G[a].push_back(b);  \
        G[b].push_back(a);  \
    }while(0);      

inline void add(int a, int b) {
    G[a].push_back(b);
    G[b].push_back(a);
}

int main() {
    std::ios::sync_with_stdio(false);
    cin>>n;
    for (int i = 1; i <= n; i++)
        cin>>c[i];
    for (int i = 1; i < n; i++) {
        int u, v;
        cin>>u>>v;
        // !! AC !!
        G[u].push_back(v);
        G[v].push_back(u);

        // !! inline or macro will TLE !!
        //add(u, v); add(v, u);
        //ADD(u, v); ADD(v, u);
    }
    dfs(1, 0);
    for (int i = 1; i <= n; i++) {
        cout<< cal(n) - ans[i] - cal(n - cnt[i])<<endl;
    }
    return 0;
}
...