題目意思
相信大家都學過樹塔問題,題目很簡單,求最大化一個三角形數塔從上往下走的路徑和。走的規則是:(i,j)號點只能走向(i+1,j)或者(i+1,j+1)。如下圖是一個數塔,映射到該數塔上行走的規則爲:從左上角的點開始,向下走或向右下走直到最底層結束。
1
3 8
2 5 0
1 4 3 8
1 4 2 5 0
路徑最大和是1+8+5+4+4 = 22,1+8+5+3+5 = 22或者1+8+0+8+5 = 22。
小S覺得這個問題so easy。於是他提高了點難度,他每次ban掉一個點(即規定哪個點不能經過),然後詢問你不走該點的最大路徑和。
當然他上一個詢問被ban掉的點過一個詢問會恢復(即每次他在原圖的基礎上ban掉一個點,而不是永久化的修改)
n<=1000,m<=5*10^5
時間限制 1s
空間限制 256M
解題思路
預處理出每個點(i,j)向上走和向下走的最大和a、b,那麼a+b就是經過(i,j)的最大路徑和。這樣以後再求出每一層中,每個點的a+b的最大值和次大值,當一個點被ban掉,就可以用同一層裏面的最大值或次大值得出答案。
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 1006
#define fr(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
void read(int &x)
{
char c;
x=0;
for(c=getchar();c<'0' || c>'9';c=getchar());
for(;c>='0' && c<='9';x=x*10+c-'0',c=getchar());
}
int i,j,n,m,x,y,a[maxn][maxn],f[maxn][maxn],g[maxn][maxn],d1[maxn],jl[maxn],d2[maxn];
void pre()
{
int i,j;
for(i=n;i;i--)
fr(j,1,i)
f[i][j]=max(f[i+1][j],f[i+1][j+1])+a[i][j];
fr(i,1,n)
fr(j,1,i)
{
g[i][j]=g[i-1][j-1]+a[i][j];
if (i!=j) g[i][j]=max(g[i][j],g[i-1][j]+a[i][j]);
}
fr(i,1,n)
{
fr(j,1,i)
{
int t=f[i][j]+g[i][j]-a[i][j];
if (t>d1[i])
{
d2[i]=d1[i];
d1[i]=t,jl[i]=j;
} else if (t>d2[i])
d2[i]=t;
}
}
return;
}
int main()
{
scanf("%d%d",&n,&m);
fr(i,1,n)
fr(j,1,i) read(a[i][j]);
pre();
fr(i,1,m)
{
scanf("%d%d",&x,&y);
if (x==1 && y==1) printf("-1\n");
else if (jl[x]==y) printf("%d\n",d2[x]);
else printf("%d\n",d1[x]);
}
return 0;
}