摘花生【模擬】

02:摘花生

總時間限制: 1000ms     內存限制: 65536kB

描述

魯賓遜先生有一隻寵物猴，名叫多多。這天，他們兩個正沿着鄉間小路散步，突然發現路邊的告示牌上貼着一張小小的紙條：“歡迎免費品嚐我種的花生！——熊字”。
魯賓遜先生和多多都很開心，因爲花生正是他們的最愛。在告示牌背後，路邊真的有一塊花生田，花生植株整齊地排列成矩形網格（如圖1）。有經驗的多多一眼就能看出，每棵花生植株下的花生有多少。爲了訓練多多的算術，魯賓遜先生說：“你先找出花生最多的植株，去採摘它的花生；然後再找出剩下的植株裏花生最多的，去採摘它的花生；依此類推，不過你一定要在我限定的時間內回到路邊。”

我們假定多多在每個單位時間內，可以做下列四件事情中的一件：
1) 從路邊跳到最靠近路邊（即第一行）的某棵花生植株；
2) 從一棵植株跳到前後左右與之相鄰的另一棵植株；
3) 採摘一棵植株下的花生；
4) 從最靠近路邊（即第一行）的某棵花生植株跳回路邊。
現在給定一塊花生田的大小和花生的分佈，請問在限定時間內，多多最多可以採到多少個花生？注意可能只有部分植株下面長有花生，假設這些植株下的花生個數各不相同。
例如在圖2所示的花生田裏，只有位於(2, 5), (3, 7), (4, 2), (5, 4)的植株下長有花生，個數分別爲13, 7, 15, 9。沿着圖示的路線，多多在21個單位時間內，最多可以採到37個花生。

輸入

輸入的第一行包括一個整數T，表示數據組數
每組輸入的第一行包括三個整數，M, N和K，用空格隔開；表示花生田的大小爲M * N（1 <= M, N <= 50），多多采花生的限定時間爲K（0 <= K <= 1000）個單位時間。接下來的M行，每行包括N個非負整數，也用空格隔開；第i + 1行的第j個整數Pij（0 <= Pij <= 500）表示花生田裏植株(i, j)下花生的數目，0表示該植株下沒有花生。

輸出

輸出包括T行，每一行只包含一個整數，即在限定時間內，多多最多可以採到花生的個數。

樣例輸入

6 7 21
0 0 0 0 0 0 0
0 0 0 0 13 0 0
0 0 0 0 0 0 7
0 15 0 0 0 0 0
0 0 0 9 0 0 0
0 0 0 0 0 0 0

樣例輸出

37

---

#include <bits/stdc++.h>
using namespace std;

const int L=55;
int a[L][L];

int main(){
	int w;scanf("%d",&w);
	while(w--){
		int m,n,t;
		scanf("%d%d%d",&m,&n,&t);
		priority_queue<int> q;
		for(int i=0;i<m;i++){
			for(int j=0;j<n;j++){
				scanf("%d",&a[i][j]);
				q.push(a[i][j]);
			}
		}
		int sum=0;
		int x0_i=0,x0_j=0;
		int t_fristPush=0,t_fristBack=0;
		int x0=q.top();q.pop();
		if(x0==0){
			printf("%d\n",sum);
			continue;
		}
		for(int i=0;i<m;i++){
			for(int j=0;j<n;j++){
				if(a[i][j]==x0){
					x0_i=i;
					x0_j=j;
					t_fristPush=i+2;
					t_fristBack=i+1;
				}
			}
		}
		if(t<t_fristPush+t_fristBack){
			printf("%d\n",sum);
			continue;
		}
		else if(t==t_fristPush+t_fristBack){
			sum+=x0;
			printf("%d\n",sum);
			continue;
		}
		else{
			sum+=x0;
			t-=t_fristPush;
			int t_push=0,t_back=0;
			while(t>0){
				int x_i=0,x_j=0;
				int x;
				if(!q.empty()){
					x=q.top();
					q.pop();
				}
				else{
					x=0;
				}
				if(x==0){
					printf("%d\n",sum);
					break;
				}
				for(int i=0;i<m;i++){
					for(int j=0;j<n;j++){
						if(a[i][j]==x){
							x_i=abs(x0_i-i);
							x_j=abs(x0_j-j);
							x0_i=i;
							x0_j=j;
							t_push=x_i+x_j+1;
							t_back=i+1;
							break;
						}
					}
				}
				if(t_push+t_back>t){
					printf("%d\n",sum);
					break;
				}
				else if(t_push+t_back==t){
					sum+=x;
					printf("%d\n",sum);
					break;
				}
				else{
					sum+=x;
					t-=t_push;
					t_push=0;
					t_back=0;
				}
			}
		}
	}
	return 0;
}