51 Nod1624 取余最长路

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

#include<iostream>  
#include<algorithm>  
#include<cstdio>  
#include<queue>  
#include<map>  
#include<vector>  
#include<cstring>  
#include<cmath>  
#include<set>  
using namespace std;  
typedef long long ll;  
int INF = 0x3f3f3f3f;  
const int N = 1e5 + 10;  
int a[4][N], sum[4][N];  
int p, sb[N*10];  
int main()  
{  
    int n;  
    scanf("%d%d", &n, &p);  
    memset(sum, 0, sizeof(sum));  
    for(int i = 1; i<=3; i++)  
    {  
        for(int j = 1; j<=n; j++)  
            scanf("%d", &a[i][j]), sum[i][j] = (sum[i][j-1] + a[i][j])%p;  
    }  
    int s = 0, ans = 0;  
    set<int>S;  
    for(int i = 1; i<=n; i++)  
    {  
        s = (sum[1][i] - sum[2][i-1] + p)%p;  
        S.insert(s);  
        s = (sum[2][i] + sum[3][n] - sum[3][i-1] + p)%p;  
        ans = max(ans, (s + *(--S.end()))%p);  
        set<int>::iterator it = S.lower_bound(p - s);  
        if(it != S.begin())  
        {  
            ans = max(ans, (s + *(--it))%p);  
        }  
    }  
    cout<<ans<<endl;  
    return 0;  
}  

看的别人的题解 Orz