51nod-1624 取余最长路

原题链接

1624 取余最长路

基准时间限制：1 秒 空间限制：131072 KB 分值: 40  难度：4级算法题

 收藏

 关注

佳佳有一个n*m的带权矩阵，她想从(1,1)出发走到(n,m)且只能往右往下移动，她能得到的娱乐值为所经过的位置的权的总和。

有一天，她被下了恶毒的诅咒，这个诅咒的作用是将她的娱乐值变为对p取模后的值，这让佳佳十分的不开心，因为她无法找到一条能使她得到最大娱乐值的路径了！

她发现这个问题实在是太困难了，既然这样，那就只在3*n的矩阵内进行游戏吧！

现在的问题是，在一个3*n的带权矩阵中，从(1,1)走到(3,n)，只能往右往下移动，问在模p意义下的移动过程中的权总和最大是多少。

样例解释：

移动的方案为“下下右”。

Input

单组测试数据
第一行两个数n(1<=n<=100000),p(1<=p<=1000000000)。
接下来3行，每行n个数，第i行第j列表示a[i][j]表示该点的权(0<=a[i][j]<p)。

Output

一个整数表示答案。

Input示例

2 3
2 2
2 2
0 1

Output示例

2

num数组代表矩阵， sum数组代表num数组每一维的前缀和, sum[0][i] = num[0][1] + ..num[0][i]

佳佳走过的路径必为(0, 1) -> (0, L) -> (1, L) -> (1, R) -> (2, R) - > (2, n).定义一个set<> s,

枚举R from1 to n, (0, 1)到(0, R)的距离为sum[0][R],  s.insert(sum[0][R] - sum[1][R-1])

设d为(1, 1) 到(2, n)经过转折点(1, R)的长度 d = sum[1][R] + sum[2][n] - sum[2][r-1]; d %= p;再在s集合中找一个数使之加上d后对p取模达到最大值，更新ans(最大娱乐值)

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
#define maxn 100005
using namespace std;
typedef long long ll;

ll num[3][maxn], sum[3][maxn];
int main(){
	
//	freopen("in.txt", "r", stdin);
	int n, p;
	
	scanf("%d%d", &n, &p);
	for(int i = 0; i < 3; i++)
	 for(int j = 1; j <= n; j++){
	 	scanf("%I64d", &num[i][j]);
	 	(sum[i][j] = sum[i][j-1] + num[i][j]) %= p;
	 }
	set<ll> s;
	set<ll> ::iterator iter;
	ll ans = 0;
	for(int i = 1; i <= n; i++){
		ll k = (sum[0][i] - sum[1][i-1] + p) % p;
		s.insert(k);
		ll d = ((sum[1][i] + sum[2][n] - sum[2][i-1]) % p + p)% p;
		ans = max(ans, (*(--s.end()) + d) % p);
		iter = s.lower_bound(p - d);
		if(iter != s.begin()){
			ans = max(ans, (*(--iter) + d) % p);
		}
	}
	printf("%I64d\n", ans);
	
	return 0;
}