[CSP-S模拟测试]:毛二琛(DP)

题目描述

  $MYC$在$NOI2018$中,遇到了$day1T2$这样一个题,题目是让你求有多少“好”的排列。$MYC$此题没有获得高分,感到非常惭愧,于是回去专心研究排列了。如今数排列的题对$MYC$来说已经是小菜一碟了。于是$MYC$想考考你,扔给你了一个非常$naive$的数排列题给你。
  给定一个${0,1,2,3,...,n-1}$的排列$p$。一个${0,1,2,...,n-2}$的排列$q$被认为是优美的排列,当且仅当$q$满足下列条件:
  对排列$s={0,1,2,3,...,n-1}$进行$n–1$次交换。
  $1.$交换$s[q_0],s[q_0+1]$
  $2.$交换$s[q_1],s[q_1+1]$
  ...
  最后能使得排列$s=p$。
  问有多少个优美的排列,答案对$10^9+7$取模。

原题见:$SRM517-600$


输入格式

第一行一个正整数$n$。
第二行$n$个整数代表排列$p$。


输出格式

仅一行表示答案。


样例

样例输入:

3
1 2 0

样例输出:

1


数据范围与提示

样例解释:

$q={0,1}{0,1,2} ightarrow{1,0,2} ightarrow{1,2,0}$
$q={1,0}{0,1,2} ightarrow{0,2,1} ightarrow{2,0,1}$

数据范围:

$20\%$:$nleqslant 10$
$50\%$:$nleqslant 50$
$70\%$:$nleqslant 300$
$100\%$:$nleqslant 5,000$


题解

题目可以转化为:一个大小为$n-1$的排列,某些地方限制了相邻两数的大小关系,求方案数。

考虑$DP$,设$dp[i][j]$表示进行到了第$i$个数,第$i$个数在前$i$个数中是第$j$小的方案数。

可以预处理出来哪些位置需要往左或右移即可。

注意一些限制,以向左移为例,第$i$次交换的位置要在第$i-1$次交换之前,反之同理。

这样做出来时间复杂度是$Theta(n^3)$的,前缀和优化即可。

因为数据点没有给不满足的情况,所以下面代码中没有判不满足的情况,即$pos_i=i$。

时间复杂度:$Theta(n^2)$。

期望得分:$100$分。

实际得分:$100$分。


代码时刻

#include<bits/stdc++.h>
using namespace std;
const int mod=1000000007;
int n;
int a[5001];
bool com[5001];
long long dp[5001][5001],g[5001][5001];
long long ans;
int main()
{
	scanf("%d",&n);
	for(int i=0;i<n;i++)
		scanf("%d",&a[i]);
	for(int i=0;i<n;i++)
		if(i<a[i]){com[i-1]=1;com[a[i]-1]=1;}
		else for(int j=a[i];j<i-1;j++)com[j]=1;
	dp[0][1]=g[0][1]=1;
	for(int i=1;i<n-1;i++)
		for(int j=1;j<=i+1;j++)
		{
			if(com[i-1])dp[i][j]=(dp[i][j]+g[i-1][i]-g[i-1][j-1]+mod)%mod;
			else dp[i][j]=(dp[i][j]+g[i-1][j-1])%mod;
			g[i][j]=(g[i][j-1]+dp[i][j])%mod;
		}
	for(int i=1;i<n;i++)ans=(ans+dp[n-2][i])%mod;
	printf("%lld",ans);
	return 0;
} 

rp++

原文地址:https://www.cnblogs.com/wzc521/p/11670185.html