首页 > 其他分享 >CF1503E 2-Coloring

CF1503E 2-Coloring

时间:2024-08-17 10:16:04浏览次数:15  
标签:Coloring 前缀 sum times 1LL CF1503E ll mod

CF1503E 2-Coloring

cjx 组合强。

思路

观察一下题目,不难发现只有当黄色形成如下的单峰时才合法。

染错色了,将就一下

其中两座峰的峰顶高度相加等于 \(m\),为了方便统计,我们钦定右边的峰一定在左峰下方的行出现,最后答案乘以二就是最终方案。

发现对于每一边是两个最长不下降子序列拼在一起。

设 \(f[i][j]\) 为长度为 \(i\) 的最长不下降子序列,最高点高度为 \(j\) 的方案数。

再设 \(f'[i][j]\) 为长度为 \(i\) 的最长不下降子序列,仅有一个最高点为 \(j\) 的方案数。

答案就是

\[2\times \sum_{i=1}^n\sum_{j=i+1}^{n}\sum_{k=1}^{m-1} (f[i][k]\times f'[n-i+1][k])\times (f'[j][m-k]\times f[n-j+1][m-k]) \]

相当于枚举第一个峰在行 \(i\),第二个峰在行 \(j\),第一个峰高度为 \(k\)。

变幻一下:

\[2\times \sum_{i=1}^n\sum_{k=1}^{m-1}(f[i][k]\times f'[n-i+1][k])\times \sum_{j=i+1}^{n} (f'[j][m-k]\times f[n-j+1][m-k]) \]

这样给后一个和式做一个后缀和优化,即可把复杂度优化到 \(O(n^2)\) 级别。

当然有例外是这个样子。

两峰交在一起(必须保证在连续的行且峰顶高度相加大于 \(m\)),也是一种方案。

\[2\times \sum_{i=1}^n\sum_{x=1}^{m-1}\sum_{y=1}^{m-1} (f[i][x]\times \sum_{k=1}^{y-1} f[n-i][k])\times (f[n-i][y]\times\sum_{k=1}^{x-1}f[i][k]) \]

相当于枚举左边的峰顶在行 \(i\),左峰高度为 \(x\),右峰高度为 \(y\),同时求和了峰过后的一段不下降子序列的方案。

\[2\times \sum_{i=1}^n\sum_{x=1}^{m-1} (f[i][x]\times\sum_{k=1}^{x-1}f[i][k] )\sum_{y=1}^{m-1} (f[n-i][y]\times\sum_{k=1}^{y-1} f[n-i][k]) \]

对于 \(\sum_{k=1}^{x-1}f[i][k]\) 和 \(\sum_{k=1}^{y-1} f[n-i][k]\) 前缀和优化。

然后预处理出所有的

\[f[n-i][y]\times\sum_{k=1}^{y-1} f[n-i][k] \]

再次前缀和。

即可在 \(O(n^2)\) 的时间内求出上述和式。

对于 \(f\) 和 \(f'\) 的转移是容易的。

\[f[i][j]=\sum_{k=1}^j f[i-1][k] \]

\[f'[i][j]=\sum_{k=1}^{j-1} f[i-1][k] \]

初始值 \(f[0][0]=1\)。

其实你看到这里,已经可以使用 dp 做出本题了,但是如果你想进一步了解组合,请移步。

不难发现,我们每次的转移都相当于求一次前缀和,而 \(f[0][0]=1\)。

所以有可以构造这样的生成函数:

\[f[0]=1 \]

观察 \(\sum_{i=0} x^i\) 的优雅性质,你会发现 \(1\times \sum_{i=0} x^i\) 相当于求一次前缀和。

而 \((\sum_{i=0} x^i)^2\) 相当于求两次前缀和(\(x^i\) 的系数是第 \(i\) 项的求前缀和后的值)。

由 \(f[0]\) 转移到 \(f[1]\) 的过程相当于乘以 \(\sum_{i=0} x^i\)(一次前缀和),所以有:

\[f[1][j]=[x^j](\sum_{k=0} x^k)^1 \]

推广一下,每次 \(f[i]\) 向 \(f[i+1]\) 的转移都是乘以 \(\sum_{i=0} x^i\)(每转移一次求一次前缀和),所以有:

\[f[i][j]=[x^j](\sum_{k=0}x^k)^i \]

对于 \(\sum_{i=0} x^i\),的封闭形式为 \((1-x)^{-1}\) 有很多种方法求,这里不再赘述。

所以有:

\[f[i][j]=[x^j](1-x)^{-i} \]

使用广义二项式展开 \((1+(-x))^{-i}\),即可得到第 \(j\) 项的系数为:

\[\binom{-i}{j}=(-1)^j\binom{i+j-1}{j} \]

关于牛顿二项式系数,已经有完备的公式,本处直接套公式即可,故不再展开讨论。

把系数乘上 \((-x)^j\),有

\[f[i][j]=\binom{i+j-1}{j} \]

从比较简单的组合意义考虑,有 \(j\) 个球,有 \(i-1\) 个挡板,可以为空,两个挡板之间的球数相当于相邻两项的差。

那么有:

\[f[i][j]=\binom{i+j-1}{i-1}=\binom{i+j-1}{j} \]

在观察一下 \(f'\) 求值的式子,你会发现 \(f'[i][j]=f[i][j-1]\)。

所以其实你也求出 \(f'\) 的组合意义了。

相当于有 \(j\) 个球,\(i-1\) 个挡板,最后一个挡板至少挡一个球,那么就可以投入 \(j-1\) 个球供所有挡板挡。

\[f'[i][j]=\binom{i+j-2}{j-1} \]

如果考虑生成函数的话,\(f'[i][j]\) 相当于较 \(f[i][j]\) 向后平移了一个位置,那么可以写成:

\[f'[i][j]=[x^{j-1}](1-x)^{-i} \]

\[f'[i][j]=[x^j](1-x)^{-i}x \]

于是你就可以 愉快 的 AC 本题了。

CODE

#include<bits/stdc++.h>
using namespace std;

#define ll long long
#define N 5000
#define mod 998244353

const int maxn=5e3+5;

ll n,m,ans;
ll fac[maxn],inv[maxn],f[maxn];

inline ll ksm(ll x,ll y)
{
    ll sum=1;
    for(;y;y/=2,x=x*x%mod) if(y&1) sum=sum*x%mod;
    return sum;
}
inline ll C(int n,int m)
{
    if(n<m||m<0) return 0;
    return 1ll*fac[n]*inv[m]%mod*inv[n-m]%mod;
}

int main()
{
    freopen("magic.in","r",stdin);
    freopen("magic.out","w",stdout);
    for(int i=fac[0]=1;i<=N;i++) fac[i]=fac[i-1]*i%mod;
    inv[N]=ksm(fac[N],mod-2);
    for(int i=N-1;~i;i--) inv[i]=inv[i+1]*(i+1)%mod;
    scanf("%lld%lld",&n,&m);
    for(int h=1;h<n;h++)
    {
        for(int j=m;j>=2;j--)
        {
            f[j]=(1LL*f[j+1]+1LL*C(j+n-h-2,n-h-1)*C(m-j+n-h,n-h)%mod)%mod;
        }
        for(int i=m-1;i>=1;i--)
        {
            ans = (1LL*ans+1LL*C(i+h-1,h)*C(m-i+1+h-2,h-1)%mod*f[i+1]%mod)%mod;
        }
    }
    memset(f,0,sizeof(f));
    for(int x=1;x<=m;x++)
    {
        for(int j=n-1;j>=1;j--)
        {
            f[j]=(1LL*f[j+1]+1LL*C(m-x+n-j-1,n-j-1)*C(m-x+j-1,j)%mod)%mod;
        }
        for(int i=1;i<n;i++)
        {
            ans=(1LL*ans+1LL*C(x+i-1,i)*C(x+n-i-1,n-i-1)%mod*f[n-i+1]%mod)%mod;
        }
    }
    printf("%lld\n",ans*2%mod);
}

标签:Coloring,前缀,sum,times,1LL,CF1503E,ll,mod
From: https://www.cnblogs.com/binbinbjl/p/18364095

相关文章

  • CF1503E 2-Coloring
    CF1503E2-Coloring题目大意略过。做法解析不会组合,使用了DP,但其实本质相同。我们假设所有的格子都是蓝色的,然后考虑将一些格子换成黄色的。我们考虑从每一行的两头开始将格子换成黄色,只要不把整一行都换成黄色的我们就可以保证每一行恰好有一段蓝色的格子。为了保证每一......
  • AT_agc025_b RGB Coloring 题解
    ProblemSolution由于涂绿色的得分为\(A+B\),所以可以将红色与蓝色独立考虑。依次枚举红色的个数,假定为\(i\),所以剩余需要的得分为\(K-i\timesA\),判断是否能被\(B\)整除,若能,则蓝色个数为\(\frac{K-i\timesA}{B}\),设为\(j\),则总方案累加\(C^{i}_{n}\timesC^{j}_{n}\),除......
  • Interactive Game with Coloring
    看这篇题解解释一下,如果存在一个点通向父亲的边和某一条通向儿子的边的颜色相同,那么一开始如果起点就在这个点的话,是没有办法向上走的,所以任意一个点通向父亲的边和某一条通向儿子的边的颜色不同对于非特殊点,我们只要一直走没有重复出现的那个颜色就好了如果某一时刻走到了特殊......
  • D. Coloring Brackets
    原题链接题解首先,假设当前\(s(l,r)\)括号序列为合法序列,则有如下几种情况:\(l+1==r\)()\(match[r]==l\)(...)\(match[r]!=l\)(...)...(...)code#include<bits/stdc++.h>usingnamespacestd;constlonglongMOD=1e9+7;longlongdp[705][705][3][3]......
  • D. Cross Coloring
    原题链接题解设想每一个\(x,y\)代表中控台,中控台颜色改变它控制的颜色也会跟着改变,我们倒过来求,这样就能确保每个中控台有没有控制的颜色code#definelllonglong#include<bits/stdc++.h>usingnamespacestd;constllmod=998244353;llxs[200005]={0};llys[200005]......
  • Coloring Edges
    \(Solution\)link一个经典结论是有向图中的任意一个环总能由一条生成树上的从祖先到儿子的链以及一条返祖边组成,正确性显然。不妨将所有树边和横插边都染成黑色,返祖边染成白色,这样就可以保证任意一个环都有两种颜色了。判断横插边和返祖边可以用栈来维护。#include<bits/std......
  • CF111D Petya and Coloring 题解
    很明显这是一道组合题。首先特判一下,当\(m=1\)时,答案就是\(k^n\)。对于\(m>1\)的情况,我们可以得出一个结论:对于沿格子的线穿过的任何垂直线,会将棋盘分成两个非空的部分,这两个部分中的不同颜色的数量相同且总是不变。设这个不同颜色的数量为\(i\),那么左边这部分的颜色一定......
  • 题解 CF1781G Diverse Coloring
    \(\texttt{link}\)题意给定一棵\(n\)个点的二叉树,现对其每个点染成黑色或白色。一种合法的染色方案满足:对于所有黑色的点,都存在白色的点与之相邻。对于所有白色的点,都存在黑色的点与之相邻。一种染色方案的权值是染成黑色的点数与染成白色的点数之差的绝对值。\(\foral......
  • Codeforces Round 734 (Div. 3)B2. Wonderful Coloring - 2(贪心构造实现)
    思路:分类讨论:当一个数字出现的次数大于等于k,那么最多有k个能被染色,当一个数字出现的次数小于k,南那么这些数字都可能被染色还有一个条件就是需要满足每个颜色的数字个数一样多,这里记出现次数小于k的所有数字的出现次数总和为sum,将所有这些数字排序后,前sum-sum%k个数字是都可以......
  • [Codeforces] CF1774B Coloring
    CF1774BColoring题意Cirno_9baka的纸条上有\(n\)个格子,他觉得空白的纸条看着有点无趣,于是想在纸条的格子上涂上\(m\)种颜色。同时,他认为第\(i\)种颜色必须要用\(a_i\)次,且每连续\(k\)个格子里涂的颜色必须互不相同。Cirno_9baka想知道有没有这样的一种涂色方案能......