【Wannafly挑战赛24E】旅行

题面

牛客

题解

首先有一个非常显然的\(dp\):我们直接把\(s\rightarrow t\)的路径抠出来然后设\(f_{i,j}\)表示到第\(i\)个点,目前余数为\(j\)的方案数。

但是这样子复杂度显然是不对的,我们想办法快速合并对于某个点\(u\),\(s\rightarrow u\),\(t\rightarrow u\)的答案。

一般这个点\(u\)都是\(lca(s,t)\)但是我们这道题有一个特别神仙的思路就是将这个点\(u\)设为\(s,t\)在点分树上的\(lca\)。

我们对一组询问,找到\(s,t\)在点分树上的\(lca\)点\(u\)并将询问离线,在点分治到\(u\)时处理询问,那么我们对于\(u\)的询问,\(u\)到所在的这个分治树中间所有点的\(dp\)值我们都是可以求出来的,这样子我们就可以合并答案了。

最后的复杂度是点分治算路径的复杂度和回答询问复杂度,为\(O(kn\log n+qk)\)。

代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
inline int gi() {
register int data = 0, w = 1;
register char ch = 0;
while (!isdigit(ch) && ch != '-') ch = getchar();
if (ch == '-') w = -1, ch = getchar();
while (isdigit(ch)) data = 10 * data + ch - '0', ch = getchar();
return w * data;
}
const int Mod = 998244353;
const int MAX_N = 2e5 + 5;
struct Graph { int to, next; } e[MAX_N << 1];
int fir[MAX_N], e_cnt;
void clearGraph() { memset(fir, -1, sizeof(fir)); e_cnt = 0; }
void Add_Edge(int u, int v) { e[e_cnt] = (Graph){v, fir[u]}, fir[u] = e_cnt++; }
int N, K, Q, a[MAX_N];
int Root, Siz, Rmx, size[MAX_N], dep[MAX_N], fa[MAX_N];
bool used[MAX_N];
void getRoot(int x, int fa) {
size[x] = 1;
int mx = 0;
for (int i = fir[x]; ~i; i = e[i].next) {
int v = e[i].to; if (v == fa || used[v]) continue;
getRoot(v, x);
size[x] += size[v];
mx = max(mx, size[v]);
}
mx = max(Siz - size[x], mx);
if (mx < Rmx) Rmx = mx, Root = x;
}
struct Query { int x, y, id; } ;
vector<Query> vec[MAX_N];
int f[MAX_N][50], ans[MAX_N];
void dfs(int x, int fa) {
for (int i = fir[x]; ~i; i = e[i].next) {
int v = e[i].to; if (v == fa || used[v]) continue;
for (int j = 0; j < K; j++) f[v][j] = f[x][j];
for (int j = 0; j < K; j++) (f[v][(j + a[v]) % K] += f[x][j]) %= Mod;
dfs(v, x);
}
}
void Div(int x, int op) {
used[x] = 1;
if (op) {
for (int i = 0; i < K; i++) f[x][i] = 0;
f[x][0] = 1;
dfs(x, 0);
for (auto i : vec[x]) {
int tmp[50];
for (int j = 0; j < K; j++) tmp[j] = f[i.x][j];
for (int j = 0; j < K; j++) (tmp[(j + a[x]) % K] += f[i.x][j]) %= Mod;
for (int j = 0; j < K; j++)
ans[i.id] = (ans[i.id] + 1ll * tmp[j] * f[i.y][(K - j) % K]) % Mod;
}
}
for (int i = fir[x]; ~i; i = e[i].next) {
int v = e[i].to; if (used[v]) continue;
Siz = Rmx = size[v];
getRoot(v, x);
dep[Root] = dep[x] + 1;
fa[Root] = x;
Div(Root, op);
}
}
int LCA(int x, int y) {
while (x != y) {
if (dep[x] < dep[y]) swap(x, y);
x = fa[x];
}
return x;
} int main () {
clearGraph();
N = gi(), K = gi();
for (int i = 1; i < N; i++) {
int u = gi(), v = gi();
Add_Edge(u, v), Add_Edge(v, u);
}
Siz = Rmx = N;
getRoot(1, 0);
Div(Root, 0);
for (int i = 1; i <= N; i++) a[i] = gi() % K;
Q = gi();
for (int i = 1; i <= Q; i++) {
int x = gi(), y = gi();
vec[LCA(x, y)].push_back((Query){x, y, i});
}
memset(used, 0, sizeof(used));
Siz = Rmx = N;
getRoot(1, 0);
Div(Root, 1);
for (int i = 1; i <= Q; i++) printf("%d\n", ans[i]);
return 0;
}

【Wannafly挑战赛24E】旅行的更多相关文章

  1. Wannafly挑战赛25游记

    Wannafly挑战赛25游记 A - 因子 题目大意: 令\(x=n!(n\le10^{12})\),给定一大于\(1\)的正整数\(p(p\le10000)\)求一个\(k\)使得\(p^k|x\ ...

  2. Wannafly挑战赛27

    Wannafly挑战赛27 我打的第一场$Wannafly$是第25场,$T2$竟然出了一个几何题?而且还把我好不容易升上绿的$Rating$又降回了蓝名...之后再不敢打$Wannafly$了. 由 ...

  3. Wannafly 挑战赛 19 参考题解

    这一次的 Wannafly 挑战赛题目是我出的,除了第一题,剩余的题目好像对大部分算法竞赛者来说好像都不是特别友好,但是个人感觉题目质量还是过得去的,下面是题目链接以及题解. [题目链接] Wanna ...

  4. Wannafly挑战赛21A

    题目链接 Wannafly挑战赛21A 题解 代码 #include <cstdio> #include <cmath> #define MAX 1000005 #define ...

  5. Wannafly挑战赛24游记

    Wannafly挑战赛24游记 A - 石子游戏 题目大意: A和B两人玩游戏,总共有\(n(n\le10^4)\)堆石子,轮流进行一些操作,不能进行下去的人则输掉这局游戏.操作包含以下两种: 把石子 ...

  6. Wannafly挑战赛25C 期望操作数

    Wannafly挑战赛25C 期望操作数 简单题啦 \(f[i]=\frac{\sum_{j<=i}f[j]}{i}+1\) \(f[i]=\frac{f[i]}{i}+\frac{\sum_{ ...

  7. Wannafly挑战赛18B 随机数

    Wannafly挑战赛18B 随机数 设\(f_i\)表示生成\(i\)个数有奇数个1的概率. 那么显而易见的递推式:\(f_i=p(1-f_{i-1})+(1-p)f_{i-1}=(1-2p)f_{ ...

  8. Wannafly挑战赛22游记

    Wannafly挑战赛22游记 幸运的人都是相似的,不幸的人各有各的不幸. --题记 A-计数器 题目大意: 有一个计数器,计数器的初始值为\(0\),每次操作你可以把计数器的值加上\(a_1,a_2 ...

  9. 【Wannafly挑战赛4】F 线路规划 倍增+Kruskal+归并

    [Wannafly挑战赛4]F 线路规划 题目描述 Q国的监察院是一个神秘的组织.这个组织掌握了整个帝国的地下力量,监察着Q国的每一个人.监察院一共有N个成员,每一个成员都有且仅有1个直接上司,而他只 ...

随机推荐

  1. Web应用调用.Net Core API

    Web应用调用.Net Core API 一.新建Web Application应用: 选择Web Application 新建好之后页面如下: 二.新建Model.新建Model文件夹并建立apiM ...

  2. js图片压缩+ajax上传

    图片压缩用到了localresizeimg 地址: https://github.com/think2011/localResizeIMG 用起来比较简单 <input type="f ...

  3. pacman 命令详解

    Pacman 是一个命令行工具,这意味着当你执行下面的命令时,必须在终端或控制台中进行. 1.更新系统 在 Arch Linux 中,使用一条命令即可对整个系统进行更新:pacman -Syu 如果你 ...

  4. Linux下搭建keepalive+nginx

    一. 安装nginx(略) 二. 安装keepalive 下载http://www.keepalived.org/download.html 安装依赖包 yum install –y popt* gc ...

  5. 【SQL】各取所需 | SQL JOIN连接查询各种用法总结

    前面 在实际应用中,大多的查询都是需要多表连接查询的,但很多初学SQL的小伙伴总对各种JOIN有些迷糊.回想一下,初期很长一段时间,我常用的似乎也就是等值连接 WHERE 后面加等号,对各种JOIN也 ...

  6. 内网漫游之SOCKS代理大结局

    0×01 引言 在实际渗透过程中,我们成功入侵了目标服务器.接着我们想在本机上通过浏览器或者其他客户端软件访问目标机器内部网络中所开放的端口,比如内网的3389端口.内网网站8080端口等等.传统的方 ...

  7. android scroller用法及属性

    正文 一.结构 public class Scroller extends Object java.lang.Object android.widget.Scroller 二.概述 这个类封装了滚动操 ...

  8. 13 ARM指令集与Thumb指令集

    指令格式 ARM基本格式 <opcode>{<cond>}{S}{.W|.N}<Rd>,<Rn>{,<operand2>} opecode: ...

  9. Linux 部署 YUM 仓库

    本篇主要写了怎么搭建自定义的YUM源,在一个拥有大量本地网络的主机环境中,可以减少对外网的依赖. Server 安装 vsftp 包 [root@server ~]# yum install vsft ...

  10. Django 之restfromwork 源码---APIView 分析

    Django 之 djangorestframework的APIView分析 APIView 类中的as_view() 方法 首先 我们从视图函数入手,在urls.py 中的 URLconfig中添加 ...