A11326 | Sasha and the Wedding Binary Search Tree
时间限制1s
内存限制256MB
通过 / 提交0/0
题目描述
Having overcome all the difficulties and hardships, Sasha finally decided to marry his girlfriend. To do this, he needs to give her an engagement ring. However, his girlfriend does not like such romantic gestures, but she does like binary search trees $^{\dagger}$ . So Sasha decided to give her such a tree.
After spending a lot of time on wedding websites for programmers, he found the perfect binary search tree with the root at vertex $1$ . In this tree, the value at vertex $v$ is equal to $val_v$ .
But after some time, he forgot the values in some vertices. Trying to remember the found tree, Sasha wondered — how many binary search trees could he have found on the website, if it is known that the values in all vertices are integers in the segment $[1, C]$ . Since this number can be very large, output it modulo $998\,244\,353$ .
$^{\dagger}$ A binary search tree is a rooted binary tree in which for any vertex $x$ , the following property holds: the values of all vertices in the left subtree of vertex $x$ (if it exists) are less than or equal to the value at vertex $x$ , and the values of all vertices in the right subtree of vertex $x$ (if it exists) are greater than or equal to the value at vertex $x$ .
After spending a lot of time on wedding websites for programmers, he found the perfect binary search tree with the root at vertex $1$ . In this tree, the value at vertex $v$ is equal to $val_v$ .
But after some time, he forgot the values in some vertices. Trying to remember the found tree, Sasha wondered — how many binary search trees could he have found on the website, if it is known that the values in all vertices are integers in the segment $[1, C]$ . Since this number can be very large, output it modulo $998\,244\,353$ .
$^{\dagger}$ A binary search tree is a rooted binary tree in which for any vertex $x$ , the following property holds: the values of all vertices in the left subtree of vertex $x$ (if it exists) are less than or equal to the value at vertex $x$ , and the values of all vertices in the right subtree of vertex $x$ (if it exists) are greater than or equal to the value at vertex $x$ .
输入格式
Each test consists of multiple test cases. The first line contains a single integer $t$ ( $1 \le t \le 10^5$ ) — the number of test cases. The description of the test cases follows.
The first line of each test case contains two integers $n$ and $C$ ( $2 \leq n \leq 5 \cdot 10^5$ , $1 \leq C \leq 10^9$ ) — the number of vertices in the tree and the maximum allowed value at the vertex.
The next $n$ lines describe the vertices of the tree. The $i$ -th line contains three integers $L_i, R_i$ and $val_i$ ( $-1 \le L_i, R_i \le n$ , $-1 \le val_i \le C$ , $L_i, R_i, val_i \ne 0$ ) — the number of the left child, the number of the right child, and the value at the $i$ -th vertex, respectively. If $L_i = -1$ , then the $i$ -th vertex has no left son. If $R_i = -1$ , then the $i$ -th vertex has no right son. If $val_i = -1$ , then the value at the $i$ -th vertex is unknown.
It is guaranteed that at least one suitable binary search tree exists.
It is guaranteed that the sum of $n$ over all test cases does not exceed $5 \cdot 10^5$ .
The first line of each test case contains two integers $n$ and $C$ ( $2 \leq n \leq 5 \cdot 10^5$ , $1 \leq C \leq 10^9$ ) — the number of vertices in the tree and the maximum allowed value at the vertex.
The next $n$ lines describe the vertices of the tree. The $i$ -th line contains three integers $L_i, R_i$ and $val_i$ ( $-1 \le L_i, R_i \le n$ , $-1 \le val_i \le C$ , $L_i, R_i, val_i \ne 0$ ) — the number of the left child, the number of the right child, and the value at the $i$ -th vertex, respectively. If $L_i = -1$ , then the $i$ -th vertex has no left son. If $R_i = -1$ , then the $i$ -th vertex has no right son. If $val_i = -1$ , then the value at the $i$ -th vertex is unknown.
It is guaranteed that at least one suitable binary search tree exists.
It is guaranteed that the sum of $n$ over all test cases does not exceed $5 \cdot 10^5$ .
输出格式
For each test case, output a single integer — the number of suitable binary search trees modulo $998\,244\,353$ .
输入输出样例
输入 #1
3 5 5 2 3 -1 -1 -1 2 4 -1 3 -1 5 -1 -1 -1 -1 3 69 2 3 47 -1 -1 13 -1 -1 69 3 3 2 3 -1 -1 -1 -1 -1 -1 -1
输出 #1
4 1 10
In the first test case, the binary search tree has the following form:
Then the possible values at the vertices are: $[2, 2, 3, 2, 2]$ , $[2, 2, 3, 2, 3]$ , $[2, 2, 3, 3, 3]$ , and $[3, 2, 3, 3, 3]$ .
In the second test case, the values at all vertices are known, so there is only one suitable binary search tree.
Then the possible values at the vertices are: $[2, 2, 3, 2, 2]$ , $[2, 2, 3, 2, 3]$ , $[2, 2, 3, 3, 3]$ , and $[3, 2, 3, 3, 3]$ .
In the second test case, the values at all vertices are known, so there is only one suitable binary search tree.
C++ 编辑器
输入
输出
可保存默认模板;新题优先使用已保存模板。
当前快捷键仅展示,暂不支持修改。
- 撤销
Ctrl / ⌘ + Z - 重做
Ctrl / ⌘ + Y - 查找
Ctrl / ⌘ + F - 全选
Ctrl / ⌘ + A - 复制
Ctrl / ⌘ + C - 剪切
Ctrl / ⌘ + X - 粘贴
Ctrl / ⌘ + V - 自动排版
工具栏排版按钮 - 草稿保存
编辑时自动保存到本机
历史
提交记录
状态说明时间源码
AI
作答助手
你好,我是作答助手。可以问思路、复杂度、样例含义或代码报错原因;不会直接给出完整 AC 代码。
确定要清空代码吗?
提交通过
评测结果:Accepted