問題 B: 【動態規劃】雷濤的小貓

時間限制: 1 Sec  記憶體限制: 128 MB
提交: 149  解決: 49
[提交] [狀態] [討論版] [命題人:admin]

題目描述

雷濤的小貓雷濤同學非常的有愛心，在他的宿舍裡，養著一隻因為受傷被救助的小貓（當然，這樣的行為是違反學生宿舍管理條例的）。  
在他的照顧下，小貓很快恢復了健康，並且愈發的活潑可愛了。可是有一天，雷濤下課回到寢室，卻發現小貓不見了！經過一番尋找，才發現她正趴在陽臺上對窗外的柿子樹發呆…在北京大學的校園裡，有許多柿子樹，在雷濤所在的宿舍樓前，就有N棵。並且這N棵柿子樹每棵的高度都是H。冬天的寒冷漸漸籠罩了大地，樹上的葉子漸漸掉光了，只剩下一個個黃澄澄的柿子，看著非常喜人。而雷濤的小貓恰好非常的愛吃柿子，看著窗外樹上的柿子，她十分眼饞，於是決定利用自己敏捷的跳躍能力跳到樹上去吃柿子。小貓可以從宿舍的陽臺上跳到窗外任意一棵柿子樹的樹頂。之後，她每次都可以在當前位置沿著當前所在的柿子樹向下跳1單位距離。當然，小貓的能力遠不止如此，她還可以在樹之間跳躍。每次她都可以從當前這棵樹跳到另外的任意一棵，在這個過程中，她的高度會下降Delta單位距離。每個時刻，只要她所在的位置有柿子，她就可以吃掉。整個“吃柿子行動”一直到小貓落到地面上為止。雷濤調查了所有柿子樹上柿子的生長情況。飽很想知道，小貓從陽臺出發，最多能吃到多少柿子？他知道寫一個程式可以很容易的解決這個問題，但是他現在懶於寫任何程式碼。於是，現在你的任務就是幫助雷濤寫一個這樣的程式。左圖是N=3，H=10，Delta=2的一個例子。小貓按照圖示路線進行跳躍，可以吃到最多的8個柿子

輸入

第一行三個整數N，H，Delta
接下來N行，每行一個整數Ni代表第i個樹上柱子的數量
接下來Ni個整數，每個整數Tij代表第i個樹的高度Tij上有一個柿子
1<=N,H<=2000
0<=Ni<=5000
1<=Delta<=N
1<=Ti<=H
輸入檔案不大於40960Kb

輸出

小貓能吃到多少柿子

樣例輸入

3 10 2
3 1 4 10
6 3 5 9 7 8 9
5 4 5 3 6 9

樣例輸出

8

卡輸入也是沒誰了。。

沒開輸入掛1300ms，開了之後750ms，白wa了三發

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 2e3+7;
int a[maxn][maxn],dp[maxn][maxn];
int idx[maxn];
void read(int &x){
    x = 0;
    int c,flag = 1;
    while(flag){
        c = getchar();
        isdigit(c) ? x = x*10+c-'0':flag^=1;
    }
}
int main(){
    register int n,h,delta,ni,t,ans;
    read(n),read(h),read(delta);
    //scanf("%d%d%d",&n,&h,&delta);
    for(int i=1;i<=n;i++){
        read(ni);
        //scanf("%d",&ni);
        if(!ni){n--,i--;}
        while(ni--){
            read(t);
            //scanf("%d",&t);
            a[i][t]++;
        }
    }
    for(register int i=h;i+1;i--){
        ans = 0;
        for(register int j=1;j<=n;j++){
            dp[j][i] = a[j][i] + dp[j][i+1];
            if(i+delta<=h)
                dp[j][i] = max(dp[j][i],dp[idx[i+delta]][i+delta]+a[j][i]);
            if(ans<dp[j][i])
                ans = dp[j][i],idx[i] = j;
        }
    }
    printf("%d\n",ans);

    return 0;
}