51Nod－算法马拉松13－A－取余最长路

佳佳有一个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 <stdio.h>
#define MAX(a, b) a > b ? a : b
#define MAXSIZE 100005

long a[4][MAXSIZE];
long b[4][MAXSIZE] = {
  0};
long p;

void init(int n)
{
    for (int i = 1; i <= 3; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            b[i][j] = a[i][j] + b[i][j - 1];
            if (b[i][j] >= p)
            {
                b[i][j] -= p;
            }
        }
    }
    return ;
}

long map(int n)
{
    long max = 0;
    long c, d;

    //将路径分为三段，每行为一段
    for (int i = 1; i < n; i++)
    {
        for (int j = i + 1; j <= n; j++)
        {
            c = b[2][j] - b[2][i - 1] > 0 ? b[2][j] - b[2][i - 1] : b[2][j] - b[2][i - 1] + p;  //第二段
            d = b[3][n] - b[3][j - 1] > 0 ? b[3][n] - b[3][j - 1] : b[3][n] - b[3][j - 1] + p;  //第三段
            max = MAX(max, (b[1][i] + c + d) % p);
        }
    }
    return max;
}

int main(int argc, const char * argv[])
{
    int n;
    scanf("%d %ld", &n, &p);
    for (int i = 1; i <= 3; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            scanf("%ld", a[i] + j);
        }
    }

    init(n);
    printf("%ld\n", map(n));

    return 0;
}

OVER！！！