Kanade Loves Maze Designing

最新推荐文章于 2021-08-27 21:01:04 发布

donname

最新推荐文章于 2021-08-27 21:01:04 发布

阅读量187

点赞数

分类专栏：题解

本文链接：https://blog.csdn.net/donname/article/details/119236050

版权

题解专栏收录该内容

48 篇文章 0 订阅

订阅专栏

本文介绍了一种解决给定无环图中f(i,19560929)模运算问题的方法，通过O(n)暴力遍历和O(n^2)暴力求解的对比，展示了如何利用预计算和递归实现更快的求解，适用于节点数量不超过5000的情况。

摘要由CSDN通过智能技术生成

题目链接

题意:给你一颗n个节点的无环图,每个点有一个值，定义 $f(i,x)=sum_{j=1}^na_{ij}x^{j-1}$ , $a_{ij}$ 表示从 $i$ 节点到 $j$ 节点路径上不同的点值个数,先在要求求出所有的 $f (i, 19560929) m o d (1 e 9 + 7)$ 和 $f (i, 19560929) m o d (1 e 9 + 9)$

解法: $\sum n \leq 5000$ , $O(n^2)$ 暴力即可, $O (n)$ 可以暴力遍历求出一个 $i$ 对应的 $f (i, 19560929) m o d (1 e 9 + 7)$ 和 $f (i, 19560929) m o d (1 e 9 + 9)$

AcCode:

#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
#include <set>

// #define inf 1<<63-1
#define int long long
#define pint std::pair<long long,long long>

const int N = 1e4 + 100;
const int mod1 = 1e9 + 7;
const int mod2 = 1e9 + 9;
const int x = 19560929;

std::vector<int> vec[N];
int value[N];
int vis[N];
int inv1[N],inv2[N];

inline int quick_pow(int a, int b, int p) {
    int ans = 1;
    a %= p;
    while (b) {
        if (b & 1) ans = ans * a % p;
        b >>= 1;
        a = a * a % p;
    }
    return ans;
}


inline pint dfs(int now, int pre, int num) {
    if (!vis[value[now]]) {
        num++;
    }
    vis[value[now]]++;
    int ans1 = inv1[now - 1];
    int ans2 = inv2[now - 1];
    ans1 = 1ll*ans1 * num % mod1;
    ans2 = 1ll*ans2 * num % mod2;
    pint res = std::make_pair(ans1, ans2);
    for (auto &v : vec[now]) {
        if (v == pre) continue;
        pint ret = dfs(v, now, num);
        res.first = (res.first + ret.first) % mod1;
        res.second = (res.second + ret.second) % mod2;
    }
    vis[value[now]]--;
    return res;
}

signed main() {
    int t; scanf("%lld", &t);
    inv1[0] = 1;
    inv2[0] = 1;
    for (int i = 1; i <= 5005; i++) {
        inv1[i] = inv1[i - 1] * x % mod1;
        //int temp = quick_pow(x, i, mod1);
        //if (temp != inv1[i]) printf("error inv1 %d\n", i);
        inv2[i] = inv2[i - 1] * x % mod2;
        //temp = quick_pow(x, i, mod2);
        //if (temp != inv2[i]) printf("error inv %d\n", i);
    }
    /*int out1 = inv1[0]%mod1 + 2 * inv1[1]%mod1 + 2 * inv1[2]%mod1 + 2 * inv1[3]%mod1 + 2 * inv1[5] % mod1;
    std::cout << out1 << std::endl;*/
    while (t--) {
        int n; scanf("%lld", &n);
        for (int i = 1; i <= n; i++) vec[i].clear();
        for (int i = 2; i <= n; i++) {
            int pre; scanf("%lld", &pre);
            vec[i].push_back(pre);
            vec[pre].push_back(i);
        }
        for (int i = 1; i <= n; i++) scanf("%lld", &value[i]);
        for (int i = 1; i <= n; i++) {
            //for (int j = 1; j <= n; j++) vis[j] = 0;
            pint ans = dfs(i, 0, 0);
            printf("%lld %lld\n", (ans.first%mod1+mod1)%mod1, (ans.second%mod2+mod2)%mod2);
        }
    }
    
}