7.1 NOI模拟赛 计数问题 dp

avatar
avatar

还是可以想出来的题目 不过考场上没有想出来 要 引以为戒。

初看觉得有点不可做 10分给到了爆搜。

考虑第一个特殊情况 B排列为1~m.

容易发现A排列中前m个数字 他们之间不能产生交换 且 第k个数字要交换到后面的m+1~n这些数字的时候 k~m的数字都要进行交换才行。

那么直接枚举有多少个数字到后面了 组合数可以解决 考虑剩下的那些空位怎么办。

其实就是要求出 (f_i) 其表示满足题目条件的i个数的排列的个数.

考虑递推 容易发现(f_0=1,f_1=1) 对于(f_i)考虑第i个数字不换 那么为(f_{i-1})换的话有i-1种方法 对应的方案为((i-1)cdot f_{i-2})

综上可以得到(f_i=f_{i-1}+(i-1)cdot f_{i-2})

这样结合上面的dfs就可以获得30分了。

考虑另外一种递推的模型 可以发现m在第一个 那么就要想办法将m换到第一个。

显然 m可以不动 那么剩下的m-1个数字就要放到后面了。

m可以直接和1交换 那么剩下的m-2个数字要放到后面。

m还可以和后面的进行交换 其实就是m个数字放到后面。

三种情况分别讨论即可。结合上面的两种方法就可以得到50分了。具体细节看代码。

上午也只推到这里。因为感觉正解比较难 所以就每有一直探索下去 其实可以继续走下去。

考虑正解:综上其实可以发现 m个数全部扔到后面是一定合法的,那么其实就是考虑有多少个数字可以被扔到前面。

如果有k个数字在后面 那么前面这s=m-k个数字就应该在前面 且他们不能和后面的k个数字存在交换。

这样就要求他们是一个合法排列 容易想到合法情况最多只有一种 这点容易证明。

也就是说此时只需要判断s个数字放到前面是否合法 即满足题目中的条件。

可以暴力安排位置 因为第一个要放到数值最小的位置上去 这样暴力放然后再check即可。

复杂度(m^2+n)

code
//#include<bitsstdc++.h>
#include<iostream>
#include<iomanip>
#include<cstdio>
#include<cstring>
#include<string>
#include<ctime>
#include<cmath>
#include<cctype>
#include<cstdlib>
#include<queue>
#include<deque>
#include<stack>
#include<vector>
#include<algorithm>
#include<utility>
#include<bitset>
#include<set>
#include<map>
#define ll long long
#define db double
#define INF 100000000000000000ll
#define ldb long double
#define pb push_back
#define put_(x) printf("%d ",x);
#define get(x) x=read()
#define gt(x) scanf("%d",&x)
#define gi(x) scanf("%lf",&x)
#define put(x) printf("%d
",x)
#define putl(x) printf("%lld
",x)
#define gc(a) scanf("%s",a+1)
#define rep(p,n,i) for(RE int i=p;i<=n;++i)
#define go(x) for(int i=lin[x],tn=ver[i];i;tn=ver[i=nex[i]])
#define fep(n,p,i) for(RE int i=n;i>=p;--i)
#define vep(p,n,i) for(RE int i=p;i<n;++i)
#define pii pair<int,int>
#define mk make_pair
#define RE register
#define P 1000000007
#define gf(x) scanf("%lf",&x)
#define pf(x) ((x)*(x))
#define uint unsigned long long
#define ui unsigned
#define EPS 1e-8
#define sq sqrt
#define mod 1000000007
#define S second
#define F first
using namespace std;
char buf[1<<15],*fs,*ft;
inline char getc()
{
    return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;
}
inline int read()
{
    RE int x=0,f=1;RE char ch=getc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getc();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getc();}
    return x*f;
}
const int MAXN=30010,maxn=10000010;
int n,m,ans;
int a[MAXN],b[MAXN];
int vis[MAXN],f[maxn],fac[maxn],inv[maxn];
inline int ksm(int b,int p)
{
	int cnt=1;
	while(p)
	{
		if(p&1)cnt=(ll)cnt*b%mod;
		b=(ll)b*b%mod;p=p>>1;
	}
	return cnt;
}
inline void dfs(int x)
{
	if(x==n+1)
	{
		int cnt=1;
		rep(1,n,i){if(a[a[i]]!=i)return;if(a[i]==b[cnt])++cnt;}
		if(cnt>m)
		{
			++ans;
			//rep(1,n,i)cout<<a[i]<<' ';
			//cout<<endl;
		}
		return;
	}
	rep(1,n,i)
	{
		if(vis[i])continue;
		a[x]=i;vis[i]=1;
		dfs(x+1);
		vis[i]=0;
	}
}
inline int C(int a,int b){return (ll)fac[a]*inv[b]%mod*inv[a-b]%mod;}
inline int check()
{
	rep(2,m,i)if(b[i]!=i-1)return 0;
	return 1;
}
inline void prepare()
{
	f[0]=1;fac[0]=1;
	rep(1,n,i)
	{
		fac[i]=(ll)fac[i-1]*i%mod;
		f[i]=(f[i-1]+(ll)f[i-2]*(i-1))%mod;
	}
	inv[n]=ksm(fac[n],mod-2);
	fep(n-1,0,i)inv[i]=(ll)inv[i+1]*(i+1)%mod;
}
int main()
{
	freopen("1.in","r",stdin);
	//freopen("ya.out","w",stdout);
	get(n);get(m);int flag=0;
	if(n<m){put(0);return 0;}
	rep(1,m,i)
	{
		get(b[i]);
		if(b[i]!=i)flag=1;
	}
	if(n<=10)
	{
		dfs(1);put(ans);
		return 0;
	}
	prepare();
	if(m==1){put(f[n]);return 0;}
	if(n==m)
	{
		rep(1,m,i)if(b[b[i]]!=i){put(0);return 0;}
		put(1);return 0;
	}
	if(!flag)
	{
		int ww=min(m,n-m);
		rep(0,ww,i)ans=(ans+(ll)C(n-m,i)*f[n-m-i])%mod;
		put(ans);return 0;
	}
	if(b[1]==m&&check())
	{
		if(n>=2*m-2)//和1交换.
			ans=(ll)f[n-2*m+2]*C(n-m,m-2)%mod;
		if(n>=2*m-1)//不动.
			ans=(ans+(ll)f[n-2*m+1]*C(n-m,m-1))%mod;
		if(n>=2*m)//交换.
			ans=(ans+(ll)f[n-2*m]*C(n-m,m))%mod;
		put(ans);
	}
	else
	{
		if(n>=2*m)ans=(ans+(ll)f[n-2*m]*C(n-m,m))%mod;
		rep(1,m,i)//枚举前m个位置
		{
			if(n<2*m-i)continue;
			int flag=0,ww=0;
			vis[b[i]]=1;
			rep(1,i,j)
			{
				++ww;
				while(!vis[ww])++ww;
				a[b[j]]=ww;
			}
			rep(1,m,j)if(vis[j]&&a[a[j]]!=j){flag=1;break;}
			if(!flag)ans=(ans+(ll)f[n-2*m+i]*C(n-m,m-i))%mod;
			//cout<<flag<<endl;
		}
		put(ans);
	}
	return 0;
}
原文地址:https://www.cnblogs.com/chdy/p/13220937.html