首页 > 其他分享 >[ABC259Ex] Yet Another Path Counting 题解

[ABC259Ex] Yet Another Path Counting 题解

时间:2024-02-22 14:25:03浏览次数:36  
标签:ifac int 题解 kMaxN ABC259Ex leq kMaxS Another fac

Description

有 \(N\) 行 \(N\) 列的网格图,只能向下或向右走,合法路径的开端和结尾的格子上数字一样

找到合法路径条数,对 \(998244353\) 取模

\(1\leq N\leq 400,1\leq a_{i,j}\leq N^2\)。

Solution

有一个 \(O(n^4)\) 的做法是每次枚举起点和终点然后用组合数计算答案,但是由于同一颜色的点可能很多所以这个做法过不了。

注意到出现次数 \(\leq n\) 的颜色显然这样做是可以的,而出现次数 \(>n\) 的颜色最多 \(n\) 个,所以对于这些出现次数很多的颜色在网格图上进行 dp 即可。

时间复杂度:\(O(n^3)\)。

Code

#include <bits/stdc++.h>

#define int int64_t

const int kMaxN = 405, kMaxS = kMaxN * kMaxN, kMod = 998244353;

int n;
int col[kMaxN][kMaxN], fac[kMaxS], ifac[kMaxS], inv[kMaxS];
std::vector<std::pair<int, int>> vec[kMaxS];

int C(int m, int n) {
  if (m < n || m < 0 || n < 0) return 0;
  return 1ll * fac[m] * ifac[n] % kMod * ifac[m - n] % kMod;
}

void prework() {
  fac[0] = ifac[0] = fac[1] = ifac[1] = inv[1] = 1;
  for (int i = 2; i <= 2 * n; ++i) {
    inv[i] = 1ll * (kMod - kMod / i) * inv[kMod % i] % kMod;
    fac[i] = 1ll * i * fac[i - 1] % kMod;
    ifac[i] = 1ll * inv[i] * ifac[i - 1] % kMod;
  }
}

int solve1(int x) {
  int ret = 0;
  for (auto p1 : vec[x]) {
    ret = (ret + 1) % kMod;
    for (auto p2 : vec[x]) {
      if (p1.first <= p2.first && p1.second <= p2.second && p1 != p2) {
        ret = (ret + C(p2.first - p1.first + p2.second - p1.second, p2.first - p1.first)) % kMod;
      }
    }
  }
  return ret;
}

int solve2(int x) {
  static int f[kMaxN][kMaxN] = {0};
  int ret = 0;
  for (int i = 1; i <= n; ++i) {
    for (int j = 1; j <= n; ++j) {
      f[i][j] = ((col[i][j] == x) + f[i - 1][j] + f[i][j - 1]) % kMod;
      if (col[i][j] == x) ret = (ret + f[i][j]) % kMod;
    }
  }
  return ret;
}

void dickdreamer() {
  std::cin >> n;
  for (int i = 1; i <= n; ++i) {
    for (int j = 1; j <= n; ++j) {
      std::cin >> col[i][j];
      vec[col[i][j]].emplace_back(i, j);
    }
  }
  prework();
  int ans = 0;
  for (int i = 1; i <= n * n; ++i) {
    if (vec[i].size() <= n) ans = (ans + solve1(i)) % kMod;
    else ans = (ans + solve2(i)) % kMod;
  }
  std::cout << ans << '\n';
}

int32_t main() {
#ifdef ORZXKR
  freopen("in.txt", "r", stdin);
  freopen("out.txt", "w", stdout);
#endif
  std::ios::sync_with_stdio(0), std::cin.tie(0), std::cout.tie(0);
  int T = 1;
  // std::cin >> T;
  while (T--) dickdreamer();
  // std::cerr << 1.0 * clock() / CLOCKS_PER_SEC << "s\n";
  return 0;
}

标签:ifac,int,题解,kMaxN,ABC259Ex,leq,kMaxS,Another,fac
From: https://www.cnblogs.com/Scarab/p/18027215

相关文章

  • 牛吃草 题解
    牛吃草居然真的是牛吃草Description由于现代化进程的加快,农场的养殖业也趋向机械化。lyz决定购置若干台自动喂草机来减少自己每天的工作量。为了简化问题,lyz决定将草地建模成一条线段,总长为\(n\),即共有\(n\)个单位长度,编号从左至右为\(1∼n\)。lyz可以在每个单位长度独......
  • blocks 单调栈、单调队列题解
    blocks题解:1、题面:2、分析:题意大概就是说,找一段最长的区间,并且这段区间的平均值>=k,那么我们可以对他的每一个值减去k,最终求和>=0即可。那我们需要对每个可能的左端点和右端点进行考虑,并以此让他们进行配对,看他们之间的区间和是否非负。那么我们先定住一个右端点,再依次考虑......
  • 理想的正方形 题解
    题目描述有一个a*b的整数组成的矩阵,现请你从中找出一个n*n的正方形区域,使得该区域所有数中的最大值和最小值的差最小。输入格式第一行为3个整数,分别表示a,b,n的值第二行至第a+1行每行为b个非负整数,表示矩阵中相应位置上的数。每行相邻两数之间用一空格分隔。100%的数据2<......
  • P5344 【XR-1】逛森林 题解
    题目链接:逛森林很早就想写写倍增优化建图,尤其是这题,奈何之前知识点没点够,本题线段树优化建图要优一些,不再赘述,没注意\(m\)是\(1e6\),挂了\(n\)多发才发现。后续再详细讲解倍增优化建图,这里简述本题做法。倍增优化建图其实和线段树优化建图恰不多的思想,为倍增求\(LCA\)的每......
  • 2024年2月21号题解
    106.从中序与后序遍历序列构造二叉树力扣题目链接解题思路找到根节点在中序序列的位置计算左子树的节点个数开辟一个节点,并把根节点的值赋值给这个节点根节点的左孩子和右孩子重复上面几个步骤代码实现/***Definitionforabinarytreenode.*structTreeNode{......
  • Day-7 模拟赛题解
    Day-7模拟赛题解S+N---【玄英计划】---2月21日---模拟测#3【补题】-比赛-梦熊联盟T1数据点3-5枚举每一个问号对应的字母Kmp,把s当作模式串匹配T\(O(26^k|T|)\),k是?的个数代码(我也不知道为啥T了,鸽着)正解有种被诈骗了的感觉根据期望的可加性,答案等于......
  • P10064 [SNOI2024] 公交线路 题解
    非常好题目。思路可以发现限制最严的一定是两个叶子的联通性。我们不妨把一个叶子向外起到联通性作用的路径称为有用的路径。也就是这个叶子走这条路径一定可以两步以内到达任意点。这个路径集合有什么作用呢。有一个性质:整个集合的路径的交最终会形成一个连通块。那么我们......
  • Atm/抢掠计划——题解
    题目描述样例671223352441266510128161514435647解析题目明显是最长路,可以用spfa求最长路,但数据范围5e5明显不允许,所以我们可以用tarjan优化一下,然后这就变成了一道tarjan板子题,先用tarjan缩点,点权为几个点之和,把所有点再存到一个数组中,再按......
  • 洛谷P4447题解
    md这篇反正不交题解的,随便写,不管它格式题意简化下,给你N个数,求出连续值分组人数最小的那组的人数最大值。这个题目还挺经典的,原本23年8月份过了到现在来又不会了(划掉,bushi对于这种,很容易想到的是输入,之后排序,然后分组这种模板就不多说了,就在我24年2月份重温这道题再打一遍代码......
  • 1 c++算法题解析-两个数之和
    //给定一个整数数组nums和一个整数目标值target,请你在该数组中找出和为目标值target的那两个整数,并返回它们的数组下标。//你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。//你可以按任意顺序返回答案。//示例1:////输入:nums=[2,7,......