51nod 1624-取余最长路（set+二分）

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

System Message  (题目提供者)

C++的运行时限为：1000 ms ，空间限制为：131072 KB  示例及语言说明请按这里

 允许其他 AC 的用户查看此代码，分享代码才能查看别人的代码并有机会获得勋章

      一开始总想着dp来搞，因为觉得和机器人走方格很像，然而怎么搞都不对，因为每个地方所求的解要取模，故一定会有后效性，所以dp是行不通的，呢怎么搞呢，题解是个很巧的方法，然而不是很好想，就是

从行数为3这个地方入手，因为行数只可能是3，所以仅有两个拐点，我们假设为x和y，呢我们可以用前缀和来求出采用当前两拐点的值，即：ans=sum[1][x]+sum[2][y]-sum[2][x-1]+sum[3][n]-sum[3][y-1];

难道我们要n^2搞？   当然不是了，我们可以事先将sum[1][x]-sum[2][x-1]的值加入到set容器里，然后枚举y的值二分查找就好啦，但是我们要知道y>=x,故不能一下子把x全加进去，呢很好办，枚举到哪一个就加进去哪一个就好啦，之前的又不会消失。。。还有一点是我们并不是找的最大的，而是找最接近p-1的，并且假如容器里没有合适的，呢一定是把容器末尾的值加上。。为什么呢？既然怎么加都不接近，但是容器里的所有值是不是都是小于等于p的，呢我们肯定不能找接近p-1的了，呢我们是不是可以找最接近2*p-1的，呢肯定最后+当前枚举的值一定是最接近2*p-1的啦。。。是不是傻

#include<set>
#include<map>       
#include<stack>              
#include<queue>              
#include<vector>      
#include<string>    
#include<math.h>              
#include<stdio.h>              
#include<iostream>              
#include<string.h>              
#include<stdlib.h>      
#include<algorithm>     
#include<functional>      
using namespace std;
typedef long long  ll;
#define inf  1000000000         
#define mod 1000000007               
#define maxn  106000    
#define PI 3.1415926  
#define lowbit(x) (x&-x)              
#define eps 1e-9    
ll sum[4][maxn], a[4][maxn];
int main(void)
{
	int n, i, j;
	ll p, ans = 0;
	scanf("%d%lld", &n, &p);
	for (i = 1;i <= 3;i++)
		for (j = 1;j <= n;j++)
		{
			scanf("%lld", &a[i][j]);
			a[i][j] %= mod;
		}
	for (i = 1;i <= 3;i++)
		for (j = 1;j <= n;j++)
			sum[i][j] = (sum[i][j - 1] + a[i][j]) % mod;
	set<ll>s;set<ll>::iterator it;
	for (i = 1;i <= n;i++)
	{
		s.insert((sum[1][i] - sum[2][i - 1] + p) % p);
		ll tmp = (sum[2][i] - sum[3][i - 1] + sum[3][n] + p) % p;
		it = s.lower_bound(p - tmp);
		if (it == s.begin())
			ans = max(ans, (tmp + (*(--s.end())))%p);
		else
			ans = max(ans, (tmp + (*(--it)))%p);
	}
	printf("%lld\n", ans);
	return 0;
}