1696. Salary for Robots

http://acm.timus.ru/problem.aspx?space=1&num=1696

代码及其注释：

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<string>
#include<queue>
#include<stack>
#include <iomanip>
using namespace std;
#define LL long long
const int INF=0x3f3f3f3f;
const int N=1005;
const int M=205;
int ans[2][M][M];//ans[l][i][j] 把从1到n每个人的薪水看成一串数组
//到第l个人时最大值为i往后必须取值>=j时最优值
int a[M][M],b[M][M];//辅助数组可以降低时间复杂度
int main()
{
//freopen("data.in","r",stdin);
int n,k,p;
while(cin>>n>>k>>p)
{
memset(ans,0,sizeof(ans));
for(int i=1;i<=k;++i)
ans[1][i][1]=1;//初始化
for(int l=2;l<=n;++l)
{
for(int i=1;i<=k;++i)
for(int j=1;j<=k;++j)
{
ans[l%2][i][j]=0;
a[i][j]=(ans[(l-1)%2][i][j]+a[i-1][j])%p;
b[i][j]=(ans[(l-1)%2][i][j]+b[i][j-1])%p;
}
for(int i=1;i<=k;++i)
for(int j=1;j<=k;++j)
{
if(i==j&&j==1)//特判
{ans[l%2][i][j]=1;continue;}
if(j>=i)//特判
{ans[l%2][i][j]=0;continue;}
//在第 l 个人处加的值可以使 i (+a[i][j]) 也可以是 j (+b[i][j])
ans[l%2][i][j]=(ans[l%2][i][j]+a[i][j]+b[i][j])%p;
}
}
int sum=0;
for(int i=1;i<=k;++i)
for(int j=1;j<=k;++j)
sum=(sum+ans[n%2][i][j])%p;
cout<<(sum+1)<<endl;
}
return 0;
}