字符串个数
今天来看看一个比较水的二维动规题,里面其实还有一些数学归纳和递推法的思想。
问题描述
在某次投票的过程中,用+和-表示对某一个人的支持与否,在黑板上写+或者-时小明发现了一个新的玩法。生成由“+”或“-”组成的字符串序列,其中“+”出现N次、“-”出现M次,如果要求形成的每一个字符串的任意前缀中“+”的个数不少于“-”的个数,现在他请编程高手的你来帮他计算出符合这样要求的字符串的个数。
输入
第一行,一个数T<=100,表示接下来共有T组数据;
从第二行开始,共有T行数据,每行包含两个整数N和M(1≤M≤N≤100)。
输出
共T行,对于每一组N和M,输出满足要求的字符串对20100501取模之后的数值。
题解
看到这样的题我们第一感觉就是DFS先枚举出答案再说。
DFS代码
#include<iostream>
using namespace std;
int i,j,m,n,a[10001],t,ti;
int readd()
{
int f=1,ans=0;
char ch=getchar();
while(ch<'0'||ch>'9')
{
if(ch=='-')
f=-1;
ch=getchar();
}
while(ch>='0'&&ch<='9')
{
ans*=10;
ans+=ch-'0';
ch=getchar();
}
return ans;
}
void dfs(int x,int y,int num)
{
if(num>m+n)
{for(int j=1;j<=m+n;j++)
cout<<a[j]<<" ";ti++;cout<<endl;return;}
for(int i=0;i<=1;i++)
{
if(i==1&&x>0)
{
a[num]=1;
if(n-x+1<m-y)
{a[num]=0;continue;}
else
dfs(x-1,y,num+1);
}
else if(i==0&&y>0)
{
a[num]=0;
if(n-x<m-y+1)
{a[num]=0;continue;}
else
dfs(x,y-1,num+1);
}
}
}
int main()
{
t=readd();
while(t)
{
ti=0;
t--;
n=readd(),m=readd();
a[1]=1;
dfs(n-1,m,2);
cout<<ti<<endl;
}
}
然后会有一个如图所示的表格(纵坐标表示N,横坐标表示M)
1
2 2
3 5 5
4 9 14 14
5 14 28 42 42
规律显而易见。
对于i∈[1,N](i∈N),j=1,我们有:
a[i][j]=i;
对于i∈[2,N](i∈N),j∈[1,i](j∈N),我们有:
a[i][j]=a[i-1][j]+a[i][j-1]。
这也是本题的状态转移方程。
动态规划代码
#include<iostream>
using namespace std;
int i,j,n[101],maxn,m[101],t;
int a[101][101];
int readd()
{
int f=1,ans=0;
char ch=getchar();
while(ch<'0'||ch>'9')
{
if(ch=='-')
f=-1;
ch=getchar();
}
while(ch>='0'&&ch<='9')
{
ans*=10;
ans+=ch-'0';
ch=getchar();
}
return ans;
}
int main()
{
t=readd();
for(i=1;i<=t;i++)
{
cin>>n[i]>>m[i];
maxn=max(maxn,n[i]);
}
for(i=1;i<=maxn;i++)
a[i][1]=i;
for(i=1;i<=maxn;i++)
for(j=2;j<=i;j++)
{
a[i][j]=(a[i-1][j]+a[i][j-1])%20100501;
}
for(i=1;i<=t;i++)
{
cout<<a[n[i]][m[i]]<<endl;
}
}
最后想着每算一次取一次 mod 20100501,
令人极其惊讶的是本题的数据范围,极其之小,而且这个题的时空限制是:2s,256MB。真是害怕。