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]

#include<bits/stdc++.h>
using namespace std;
const int MAXN=100100;
long long mz[5][MAXN],pre[5][MAXN];
set<long long> st;
set<long long> ::iterator it;
int main()
{
    long long n,mod,i,j,ans,x1,x2;
    while(~scanf("%lld%lld",&n,&mod))
    {
        for(i=1;i<=3;i++)
            for(j=1;j<=n;j++)
                scanf("%lld",&mz[i][j]);
        pre[1][0]=pre[2][0]=pre[3][0]=0;
        for(i=1;i<=3;i++)
            for(j=1;j<=n;j++)
                pre[i][j]=(mz[i][j]+pre[i][j-1])%mod;
        st.clear();
        ans=0;
        for(i=1;i<=n;i++)
        {
            x1=(pre[1][i]-pre[2][i-1]+mod)%mod;
            st.insert(x1);
            x2=(pre[3][n]-pre[3][i-1]+pre[2][i]+mod)%mod;
            it=st.lower_bound(mod-x2);
            ans=max(ans,(x2+*(--it))%mod);
        }
        printf("%lld\n",ans);
    }
}