【cf995】F. Cowmpany Cowmpensation(拉格朗日插值)
题意:
给出一颗树,每个结点有取值范围\([1,D]\)。
现在有限制条件:对于一个子树,根节点的取值要大于等于子数内各结点的取值。
问有多少种取值方案。
思路:
- 手画一下发现,对于一颗大小为\(sz\)的数,最终的答案为一个\(sz+1\)次为最高次幂的多项式。
- 因为节点数\(n\leq 3000\),所以暴力求出后插值即可。
简略证明:对于一个链,显然,一个长度为\(x\)的链,最终的结果为\(x+1\)次的多项式;考虑两条链的合并:长度为\(x\)的链和长度为\(y\)的链,显然两者相乘最终为\(x+y+2\)次的多项式,因为合并过后会多一个父节点,那么就是有\(x+y+1\)个点。
归纳一下就有上面说的结论了。
代码如下:
/*
* Author: heyuhhh
* Created Time: 2019/11/18 20:20:04
*/
#include <bits/stdc++.h>
#define MP make_pair
#define fi first
#define se second
#define sz(x) (int)(x).size()
#define all(x) (x).begin(), (x).end()
#define INF 0x3f3f3f3f
#define Local
#ifdef Local
#define dbg(args...) do { cout << #args << " -> "; err(args); } while (0)
void err() { std::cout << '\n'; }
template<typename T, typename...Args>
void err(T a, Args...args) { std::cout << a << ' '; err(args...); }
#else
#define dbg(...)
#endif
void pt() {std::cout << '\n'; }
template<typename T, typename...Args>
void pt(T a, Args...args) {std::cout << a << ' '; pt(args...); }
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
//head
const int N = 3005, MOD = 1e9 + 7;
ll qpow(ll a, ll b) {
ll ans = 1;
while(b) {
if(b & 1) ans = ans * a % MOD;
a = a * a % MOD;
b >>= 1;
}
return ans;
}
int n, D;
vector <int> g[N];
int res[N];
int pre[N][N];
void dfs(int u, int fa) {
int son = 0;
for(auto v : g[u]) if(v != fa) {
dfs(v, u); ++son;
}
if(!son) {
for(int i = 1; i <= n; i++) pre[u][i] = i;
} else {
for(int i = 1; i <= n; i++) res[i] = 1;
for(auto v : g[u]) if(v != fa) {
for(int i = 1; i <= n; i++) res[i] = 1ll * res[i] * pre[v][i] % MOD;
}
for(int i = 1; i <= n; i++) pre[u][i] = (pre[u][i - 1] + res[i]) % MOD;
}
}
struct Lagrange {
static const int SIZE = 3005;
ll f[SIZE], fac[SIZE], inv[SIZE], pre[SIZE], suf[SIZE];
int n;
inline void add(ll &x, int y) {
x += y;
if(x >= MOD) x -= MOD;
}
void init(int _n) {
n = _n;
fac[0] = 1;
for (int i = 1; i < SIZE; ++i) fac[i] = fac[i - 1] * i % MOD;
inv[SIZE - 1] = qpow(fac[SIZE - 1], MOD - 2);
for (int i = SIZE - 1; i >= 1; --i) inv[i - 1] = inv[i] * i % MOD;
f[0] = 0;
}
ll calc(ll x) {
if (x <= n) return f[x];
pre[0] = x % MOD;
for (int i = 1; i <= n; ++i) pre[i] = pre[i - 1] * ((x - i) % MOD) % MOD;
suf[n] = (x - n) % MOD;
for (int i = n - 1; i >= 0; --i) suf[i] = suf[i + 1] * ((x - i) % MOD) % MOD;
ll res = 0;
for (int i = 0; i <= n; ++i) {
ll tmp = f[i] * inv[n - i] % MOD * inv[i] % MOD;
if (i) tmp = tmp * pre[i - 1] % MOD;
if (i < n) tmp = tmp * suf[i + 1] % MOD;
if ((n - i) & 1) tmp = MOD - tmp;
add(res, tmp);
}
return res;
}
}lagrange;
void run(){
for(int i = 2; i <= n; i++) {
int x; cin >> x;
g[i].push_back(x);
g[x].push_back(i);
}
lagrange.init(n);
dfs(1, 0);
for(int i = 1; i <= n; i++) lagrange.f[i] = pre[1][i];
int ans = lagrange.calc(D);
cout << ans;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cout << fixed << setprecision(20);
while(cin >> n >> D) run();
return 0;
}
【cf995】F. Cowmpany Cowmpensation(拉格朗日插值)的更多相关文章
- Codeforces F. Cowmpany Cowmpensation
Description 有一棵树,现在要给每个节点赋一个在1到D之间的权值,问有多少种方案满足任意一个节点的权值都不大于其父亲的权值. n<=3000,D<=1e9 题面 Solution ...
- F. Cowmpany Cowmpensation dp+拉格朗日插值
题意:一个数,每个节点取值是1-d,父亲比儿子节点值要大,求方案数 题解:\(dp[u][x]=\prod_{v}\sum_{i=1}^xdp[v][i]\),v是u的子节点,先预处理出前3000项, ...
- 【CF995F】Cowmpany Cowmpensation(动态规划,拉格朗日插值)
[CF995F]Cowmpany Cowmpensation(多项式插值) 题面 洛谷 CF 题解 我们假装结果是一个关于\(D\)的\(n\)次多项式, 那么,先\(dp\)暴力求解颜色数为\(0. ...
- CF 622 F The Sum of the k-th Powers —— 拉格朗日插值
题目:http://codeforces.com/contest/622/problem/F 设 f(x) = 1^k + 2^k + ... + n^k 则 f(x) - f(x-1) = x^k ...
- Educational Codeforces Round 7 F - The Sum of the k-th Powers 拉格朗日插值
The Sum of the k-th Powers There are well-known formulas: , , . Also mathematicians found similar fo ...
- 【CF995F】 Cowmpany Cowmpensation
CF995F Cowmpany Cowmpensation Solution 这道题目可以看出我的代码能力是有多渣(代码能力严重退化) 我们先考虑dp,很容易写出方程: 设\(f_{i,j}\)表示以 ...
- 拉格朗日插值优化DP
拉格朗日插值优化DP 模拟赛出现神秘插值,太难啦!! 回忆拉格朗日插值是用来做什么的 对于一个多项式\(F(x)\),如果已知它的次数为\(m - 1\),且已知\(m\)个点值,那么可以得到 \[F ...
- 【CF995F】Cowmpany Cowmpensation
[CF995F]Cowmpany Cowmpensation 题面 树形结构,\(n\)个点,给每个节点分配工资\([1,d]\),子节点不能超过父亲节点的工资,问有多少种分配方案 其中\(n\leq ...
- 常系数齐次线性递推 & 拉格朗日插值
常系数齐次线性递推 具体记在笔记本上了,以后可能补照片,这里稍微写一下,主要贴代码. 概述 形式: \[ h_n = a_1 h_{n-1}+a_2h_{n-2}+...+a_kh_{n-k} \] ...
随机推荐
- easyui textbox setValue以及setText的使用技巧
1.先赋值Value,后赋值Text $('#Name').textbox('setValue', "11");$('#Name').textbox('setText', &quo ...
- 题解:P2130 狂奔的Wzf
题目链接 solution 判断有无障碍的时候不需要以此枚举,利用前缀和,如果前缀为零证明没有障碍. 重复很多,写的很丑了,#3死活不过 #include<iostream> #inclu ...
- 一天两道PAT(1)
其实是从昨天开始刷的,备战一下PAT(乙级,菜鸡不解释,希望几个月下来能有长进吧),做了一下,发现自己的算法功底好差啊..... 先上题目 1. 字符串中必须仅有P, A, T这三种字符,不可以包含其 ...
- LG1345 「USACO5.4」Telecowmunication 最小割
问题描述 LG1345 题解 点边转化,最小割,完事. \(\mathrm{Code}\) #include<bits/stdc++.h> using namespace std; tem ...
- [BZOJ1042]AOI2008]硬币购物
题目描述 Description 硬币购物一共有4种硬币.面值分别为\(c_1\) ,\(c_2\) ,\(c_3\) ,\(c_4\) .某人去商店买东西,去了\(tot\) 次.每次带\(d_i\ ...
- 新书上线:《Spring Boot+Spring Cloud+Vue+Element项目实战:手把手教你开发权限管理系统》,欢迎大家买回去垫椅子垫桌脚
新书上线 大家好,笔者的新书<Spring Boot+Spring Cloud+Vue+Element项目实战:手把手教你开发权限管理系统>已上线,此书内容充实.材质优良,乃家中必备垫桌脚 ...
- 【译】3D打印:介绍
原文地址:(需要翻墙)https://ordina-jworks.github.io/iot/2018/09/28/3D-Printing-Intro.html 文章发表日期:2018-09-28 第 ...
- Batchnorm原理详解
Batchnorm原理详解 前言:Batchnorm是深度网络中经常用到的加速神经网络训练,加速收敛速度及稳定性的算法,可以说是目前深度网络必不可少的一部分. 本文旨在用通俗易懂的语言,对深度学习的常 ...
- 2019 年 GitHub 上最热门的 Java 开源项目
1.JavaGuide https://github.com/Snailclimb/JavaGuide Star 22668 [Java 学习 + 面试指南] 一份涵盖大部分 Java 程序员所需要掌 ...
- linux的vi编辑器常用用法一览
vi 命令用于编辑文本文件,语法: vi 文件名 vi 是一个比较强大的编辑工具,类似于windows下的notepad,但是功能要强大的多.vi分为三种模式,分别是“一般模式”,“编辑模式”,“命令 ...