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;

}