AtCoder Beginner Contest 372 F题(dp)

题目链接

AtCoder Beginner Contest 372 F

思路

对于这种计数问题,我们很容易想到dp。

我们假设 d p [ i ] [ j ] dp[i][j] dp[i][j]表示第 i i i次移动,到第 j j j号节点的方案数,则状态转移方程为:

d p [ v ] [ j ] = d p [ v − 1 ] [ j − 1 ] + d p [ u ] [ j − 1 ] dp[v][j] = dp[v-1][j-1] + dp[u][j-1] dp[v][j]=dp[v−1][j−1]+dp[u][j−1]

显然,无论是时间复杂度还是空间复杂度都无法接受。

我们发现, m m m最大不会超过50,因此我们可以考虑以 m m m为突破口对dp进行优化。

我们发现,状态转移方程的前半部分实际上是对 d p dp dp数组进行循环移动

因为 d p [ v − 1 ] [ j − 1 ] dp[v-1][j-1] dp[v−1][j−1]实际上就是 d p dp dp数组循环移动一位,且 m = 0 m=0 m=0时答案为 1 1 1,因此我们可以只考虑 m m m。

我们可以转换一下,我们假设 d p dp dp数组不动,对 m m m条边进行移动,每次 m m m条边移动后的端点进行更新答案,相当于变相的移动了 d p dp dp数组。

代码

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 2e5 + 5;
const int mod = 998244353;
const int inf = 0x3f3f3f3f3f3f3f3f;
int n, m, k;
int dp[N], pre[N];
vector<pair<int, int>> edge;
int MOD(int x)
{
	return (x % n + n) % n;
}
void solve()
{
	cin >> n >> m >> k;
	for (int i = 1, u, v; i <= m; i++)
	{
		cin >> u >> v;
		u--, v--;
		edge.push_back({u, v});
	}
	dp[0] = 1;
	for (int i = 1; i <= k; i++)
	{
		for (auto [u, v] : edge)
		{
			u = MOD(u - i + 1);//上一层的u
			pre[u] = dp[u];
		}

		for (auto [u, v] : edge)
		{
			dp[MOD(v - i)] = dp[MOD(v - i)] + pre[MOD(u - i + 1)];
			dp[MOD(v - i)] %= mod;
		}
	}
	int ans = 0;
	for (int i = 0; i < n; i++)
	{
		ans = (ans + dp[i]) % mod;
	}
	cout << ans << endl;
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int test = 1;
	// cin >> test;
	for (int i = 1; i <= test; i++)
	{
		solve();
	}
	return 0;
}
相关推荐
正脉科工 CAE仿真7 分钟前
基于ANSYS 概率设计和APDL编程的结构可靠性设计分析
人工智能·python·算法
惊鸿一博20 分钟前
c++ &&(通用引用)和&(左值引用)区别
开发语言·c++
Dovis(誓平步青云)31 分钟前
【数据结构】排序算法(中篇)·处理大数据的精妙
c语言·数据结构·算法·排序算法·学习方法
2401_8729450937 分钟前
【补题】Xi‘an Invitational 2023 E. Merge the Rectangles
算法
暮雨哀尘1 小时前
微信小程序开发:开发实践
开发语言·算法·微信小程序·小程序·notepad++·性能·技术选型
nuo5342021 小时前
黑马 C++ 学习笔记
c语言·c++·笔记·学习
Touper.1 小时前
L2-003 月饼
数据结构·算法·排序算法
DARLING Zero two♡1 小时前
C++类间的 “接力棒“ 传递:继承(上)
开发语言·c++·继承·里氏替换原则
会讲英语的码农1 小时前
如何学习C++以及C++的宏观认知
开发语言·c++·学习
想跑步的小弱鸡6 小时前
Leetcode hot 100(day 3)
算法·leetcode·职场和发展