NEU- 1456: 逃票的chanming(3)

http://acm.neu.edu.cn/hustoj/problem.php?id=1456

题意很明白，但是不知道怎么下手，纠结了好久。后来看了题解，才发现是dp递推。 还是觉得自己太弱。

dp[i][j]表示在i个数里面选j个数，那么 dp[i][3]=dp[i-1][3]+dp[i-1][2]*f[i];

#include<cstdio>
#include<cstring>
#define mod 400823823
long long dp[3001][4],f[3001];   //dp[i][j]表示在i个数里面选j个数的情况
int main()
{
    int n,i;
    while(scanf("%d",&n)!=EOF)
    {
        memset(dp,0,sizeof(dp));
        for(i=1;i<=n;i++)
            scanf("%lld",&f[i]);
        dp[3][3]=(f[1]*f[2]*f[3])%mod;
        dp[3][2]=(f[1]*f[2]+f[1]*f[3]+f[2]*f[3])%mod;
        dp[3][1]=(f[1]+f[2]+f[3])%mod;
        for(i=4;i<=n;i++)
        {
            dp[i][3]=(dp[i-1][3]+dp[i-1][2]*f[i])%mod;
            dp[i][2]=(dp[i-1][2]+dp[i-1][1]*f[i])%mod;
            dp[i][1]=(dp[i-1][1]+f[i])%mod;
        }
        printf("%lld\n",dp[n][3]);
    }
    return 0;
}