牛客国庆集训派对Day6 - E Growth （DP+离散化）

https://www.nowcoder.com/acm/contest/206/E

POINT：

主要还是离散化，然后map里存，(x,y)这里属性里可以获得多少Z。

add[x][y]代表刚刚到了x，y后，每天可以获得多少z。

DP[x][y]代表刚刚到了x，y后，最多总共获得了多少z。

 

#include<stdio.h>
#include<iostream>
#include<math.h>
#include<map>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long LL;
typedef pair<int,int> pii;
const int N = 1111;

map<pii,int> mp;

int x[N],y[N],z[N];
LL add[N][N];
LL dp[N][N];


int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%d%d%d",&x[i],&y[i],&z[i]);
		mp[{x[i],y[i]}]+=z[i];
	}
	sort(x+1,x+1+n);
	sort(y+1,y+1+n);
	int tx=(int)(unique(x+1,x+1+n)-x-1);
	int ty=(int)(unique(y+1,y+1+n)-y-1);
	for(int i=1;i<=tx;i++){
		for(int j=1;j<=ty;j++){
			add[i][j]=add[i-1][j]+add[i][j-1]-add[i-1][j-1]+mp[{x[i],y[j]}];
		}
	}
	for(int i=1;i<=tx;i++){
		for(int j=1;j<=ty;j++){
			dp[i][j]=add[i][j]+max(dp[i-1][j]+(x[i]-x[i-1]-1)*add[i-1][j],dp[i][j-1]+(y[j]-y[j-1]-1)*add[i][j-1]);
		}
	}
	LL ans=0;
	for(int i=1;i<=tx;i++){
		for(int j=1;j<=ty;j++){
			if(x[i]+y[j]<=m){
				ans=max(ans,dp[i][j]+(m-x[i]-y[j])*add[i][j]);
			}
		}
	}
	printf("%lld\n",ans);

}