牛客练习赛7 C 随机树（DFS序+线段树+快速幂）

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 32768K，其他语言65536K
64bit IO Format: %lld

题目描述

平⽇⾥写hash的时候，总有某些选⼿由于脸⿊⽽导致惨遭卡模数，然后⼀些 恶意卡模数的出题⼈也因此⾝败名裂。为了防⽌被卡，我们⽤⼀种⾼级的随机 ⽅式来代替原来的线性随机⽣成，也就是所谓的随机树！ 

现在有⼀棵编号为0~n-1的有根树，其中0是树的根。每个节点初始有⼀个

值T i。现在要求⽀持⼀下两种操作：

1.给出两个正整数u和x，我们将Tu的值乘以x，我们将这种操作称为SEED 操作。 

2.给出⼀个正整数i，询问S i以及它⼀共有多少个正约数。其中S i表⽰以i为 根的⼦树所有点的权值的乘积，我们将这种操作称为RAND操作。 容易发现，这样得到的答案还是很随机的。（其实不是） 

你需要回答每⼀次的询问，由于⼀个数的约数个数可能⾮常多，这个数也 可以⾮常⼤，你只需要把答案对1e9+7取模就可以了

输入描述:

  
  

   
   第⼀⾏⼀个正整数n，表⽰节点个数。 
  
  
  
  

   
   接下来n-1⾏，每⾏两个正整数u和v，表⽰u是v的⽗节点。 
  
  
  
  

   
   接下来⼀⾏n个正整数，分别表⽰每个节点的初始权值T
   
   i。 
  
  
  
  

   
   接下来⼀⾏⼀个正整数q，表⽰操作的个数。 
  
  
  
  

   
   接下来q⾏，每⾏是以下两种情况之⼀： 
  
  
  
  

   
   1. SEED u x 表⽰将u节点的权值乘以x。 
  
  
  
  

   
   2. RAND i 表⽰询问S
   
   i以及它⼀共有多少个正约数。
  
  

输出描述:

每⼀⾏两个整数，对应⼀个RAND操作，你需要输出所求的权值以及它的 正约数个数，答案对于1e9+7取模即可。

示例1

输入

8
0 1 
0 2 
1 3 
2 4 
2 5 
3 6 
3 7
7 3 10 8 12 14 40 15 
3
RAND 1
SEED 1 13
RAND 1

输出

14400 63 
187200 126

备注:

  
  

   
   对于20%的数据，1 ≤ n, q ≤ 10。 
  
  
  
  

   
   对于40%的数据，1 ≤ n, q ≤ 100。 
  
  
  
  

   
   对于60%的数据，1 ≤ n, q ≤ 2000。 
  
  
  
  

   
   对于80%的数据，1 ≤ n, q ≤ 50000。 
  
  
  
  

   
   对于100%的数据，1 ≤ n, q ≤ 100000。 
  
  

【思路】

将树存好后深搜一遍，得到DFS序，时间戳in和out分别代表一棵子树的入口点和出口点，即可用线段树来维护、查询每个节点的各个质因数的个数，总的时间复杂度为O(nlogn)。

【代码】

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;
 
const int MAXN = 1e5 + 5;
const long long MOD = 1e9 + 7;
const long long PRIME[6] = {2, 3, 5, 7, 11, 13};
 
struct segment {
    int left, right, mid;
    int sum[6];
};
 
int n, q, cnt;
int in[MAXN], out[MAXN], value[MAXN], factor[6];
vector<int> connect[MAXN];
map<int, int> id;
segment tree[MAXN << 2];
 
void dfs(int u)
{
    in[u] = ++cnt;
    id[cnt] = u;
    for (int i = 0; i < connect[u].size(); i++) dfs(connect[u][i]);
    out[u] = cnt;
}
 
long long qpow(int a, int b)
{
    long long ans = 1;
    while (b) {
        if (b & 1) ans = ans * a % MOD;
        a = (long long)a * a % MOD;
        b >>= 1;
    }
    return ans;
}
 
void build(int left, int right, int root)
{
    tree[root].left = left;
    tree[root].right = right;
    tree[root].mid = (left + right) >> 1;
    memset(tree[root].sum, 0, 6 * sizeof(int));
    if (left == right) {
        for (int i = 0; i < 6; i++) {
            while (value[id[left]] % PRIME[i] == 0) {
                tree[root].sum[i]++;
                value[id[left]] /= PRIME[i];
            }
        }
    }
    else {
        build(left, tree[root].mid, root << 1);
        build(tree[root].mid + 1, right, root << 1 | 1);
        for (int i = 0; i < 6; i++)
            tree[root].sum[i] = tree[root << 1].sum[i] + tree[root << 1 | 1].sum[i];
    }
}
 
void modify(int left, int right, int num, int root)
{
    if (left <= tree[root].left && tree[root].right <= right) {
        for (int i = 0; i < 6; i++) {
            while (num % PRIME[i] == 0) {
                tree[root].sum[i]++;
                num /= PRIME[i];
            }
        }
    }
    else {
        if (left <= tree[root].mid)
            modify(left, right, num, root << 1);
        if (right >= tree[root].mid + 1)
            modify(left, right, num, root << 1 | 1);
        for (int i = 0; i < 6; i++)
            tree[root].sum[i] = tree[root << 1].sum[i] + tree[root << 1 | 1].sum[i];
    }
}
 
void query(int left, int right, int root)
{
    if (left <= tree[root].left && tree[root].right <= right) {
        for (int i = 0; i < 6; i++)
            factor[i] += tree[root].sum[i];
    }
    else {
        if (left <= tree[root].mid) query(left, right, root << 1);
        if (right >= tree[root].mid + 1) query(left, right, root << 1 | 1);
    }
}
 
int main()
{
    scanf("%d", &n);
    int u, v;
    for (int i = 1; i <= n - 1; i++) {
        scanf("%d %d", &u, &v);
        connect[u].push_back(v);
    }
    cnt = 0;
    dfs(0);
    for (int i = 0; i <= n - 1; i++) scanf("%lld", &value[i]);
    build(1, n, 1);
    char mes[5];
    int x;
    long long t;
    scanf("%d", &q);
    for (int k = 1; k <= q; k++) {
        scanf("%s %d", mes, &x);
        if (mes[0] == 'S') {
            scanf("%d", &t);
            modify(in[x], in[x], t, 1);
        }
        if (mes[0] == 'R') {
            memset(factor, 0, 6 * sizeof(int));
            query(in[x], out[x], 1);
            long long s = 1, ans = 1;
            for (int i = 0; i < 6; i++) {
                s = s * qpow(PRIME[i], factor[i]) % MOD;
                ans = ans * (factor[i] + 1) % MOD;
            }
            printf("%lld %lld\n", s, ans);
        }
    }
    return 0;
}