atcoder Picking Goods(DP)

题目链接
题意：
有个n*m的矩阵，里面放了k个数，现在让你从左下角走到右上角，只能向上走和向右走，路过一个数就可以把那个数加进答案，也可以不加，限制是每一行最多只能选3个数，现在问你答案最大值是多少。
r,c<=3000
k<=1e5
放进去的数<=1e9
思路：
dp[i][j][k]表示在i行j列该行已经取了k个数的最大值，dp即可。
记得开ll。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=3100;
ll r,c,tk,mat[maxn][maxn], dp[maxn][maxn][5];
int main() {
    ios::sync_with_stdio(false);cin.tie(0);cout.precision(10);cout << fixed;
#ifdef LOCAL_DEFINE
    freopen("input.txt", "r", stdin);
#endif
    memset(mat,0,sizeof(mat));
    memset(dp,0,sizeof(dp));
    cin>>r>>c>>tk;
    for(int i=0; i<tk; ++i){
        ll u,v,w;
        cin>>u>>v>>w;
        mat[u][v]=w;
    }
    dp[1][1][0]=0;
    dp[1][1][1]=mat[1][1];
    for(int i=1; i<=r; ++i){
        for(int j=1; j<=c; ++j){
            for(int k=0; k<=3; ++k){
                if(i+1<=r) {
                    dp[i+1][j][0]=max(dp[i+1][j][0], dp[i][j][k]);
                    if(mat[i+1][j]>0) dp[i+1][j][1]=max(dp[i+1][j][1], dp[i][j][k]+mat[i+1][j]);
                }
                if(j+1<=c){
                    dp[i][j+1][k]=max(dp[i][j+1][k], dp[i][j][k]);
                    if(k+1<=3 && mat[i][j+1]>0) dp[i][j+1][k+1]=max(dp[i][j+1][k+1], dp[i][j][k]+mat[i][j+1]);
                }
            }
        }
    }
    cout<<max({dp[r][c][0], dp[r][c][1], dp[r][c][2], dp[r][c][3]})<<'\n';
#ifdef LOCAL_DEFINE
    cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
#endif
    return 0;
}