题目:
写这题花了我一上午时间。
下面是本人(zhangjiuding)的思考过程:
首先想到的是三行,每一行一定要走到。
大概是这样一张图
每一行长度最少为1。即第一行(i -1) >= 1,第二行 (j - i) >= 1,第三行 (n - j) >= 1。
我们要求的就是这条路径上的和。
我们发现只要 i 和 j 固定了,结果就固定了。
如果每次都要重新求和的话时间复杂度肯定不够,所以我这个时候想到了前缀和。
用sum[a][b]表示第a行(总共只有3行)前b个数字的和。
第一行可以用 sum[1][i]来表示,第二行可以用sum[2][j]-sum[2][i-1]来表示,第三行可以用sum[3][n]-sum[3][j-1]来表示。
(此处用到了i-1所以我们吧sum[][0]空出来,这样就不会越界)
ans = sum[1][i] + sum[2][j] - sum[2][i-1] + sum[3][n] - sum[3][j-1];
每一组i、j都有一个固定结果,我们求出其中取模最大的那个就好了。
现在求和的问题解决了,该固定i , j 了。
下面看到i的取值范围(1 <= i <= n),j的取值范围(i <= j <= n)
如果遍历i和j的话时间复杂度还是O(n^2),所以前缀和+暴力肯定是会超时的。
如果能优化一点到O(n*log(n))就好了,这样就不会超时了。
那么能优化吗?答案是肯定的。
我们先把我们的每一对i,j对应的ans分离一下。
定义up = sum[1][i] - sum[2][i-1];
down = sum[2][j] + sum[3][n] - sum[3][j-1];
这样up中只含有i,down中只含有j。
因为我们知道结果只和 i , j 有关,
如果我们能把每一个up可能的down存起来,二分法找出那个能让结果最大的down值,然后用up+down更新ans就好了。
这样时间复杂度就只有O(n*log(n))了。
你可能看到这里还有疑问,先怀着疑问,后面我会把我碰到的问题和疑问都讲出来。
先看代码有助于理解:
#include <bitsstdc++.h> using namespace std; typedef long long ll; //定义结构体,value存输入的值,sum存当前行的前缀和。 struct node{ ll value;ll sum; }a[4][100001]; //ifstream in("in15.txt"); int main(){ int n; ll p; cin >> n >> p; for(int i = 1;i <= 3; i++){ for(int j = 1;j <= n; j++){ cin >> a[i][j].value; a[i][j].sum = a[i][j].value+a[i][j-1].sum; // 求前缀和。 } } ll ans = 0; set<ll> s; // 用set存当前i值可能碰到的j值对应的down。 set<ll>::iterator it; //反向遍历是因为down的范围可以慢慢扩张,不用删除set中的元素 for(int i = n; i >= 1; i--){ ll up = (a[1][i].sum-a[2][i-1].sum)%p; // 当前i值对应的up。 up = (up+p)%p; // 强行变成正数。 ll down = (a[2][i].sum+a[3][n].sum-a[3][i-1].sum)%p; // 当前i值对应的down。 down = (down+p)%p; // 强行变成正数。 // cout << "i:" << i << endl; // cout << "up:"<< up << endl; // cout << "down:" << down << endl; s.insert(down);//因为肯定是j >= i的,所以每次把当前i值对应的down加入到set,set就会表示可能取到的所有的j对应的down的集合。 //明确一点,set里装的全是down。 // 二分查找,找到一个down值与up相加后最大。 it = s.lower_bound(p-up); // 找到第一个大于等于p-up的down值,我们需要将it--,求出第一个比p-up小的。 //注意,1、2顺序不能变。 if(it != s.begin()) { //1.如果等于it = s.begin(),因为it还要减一下,相当于没有找到合适的值。 it--; // 2.it--,让it指向第一个比p-up小的值。 ans = max(ans,(up+*it)%p); // 更新 ans。 } //可能第一个比p-up小的down值加上up后还是比较小,反而down的最大值加上up后取模还比较大,所以话线性时间更新一下ans。 it = s.end(); it--; ans = max(ans,(*it+up)%p); } cout << ans << endl; return 0; } //writed by zhangjiuding
看到这里可能有一个最大的问题,那就是强行变正数的问题。
比如p = 10, up求出来的是-3,而down求出来是6,(-3 + 6)%10 = 3还是正数,强行将up变成正数是否会出错呢?
答案是肯定不会, up 强行变正后是7,(7+6)% 10 = 3,还是3。
看起来逻辑性并不强,但是我试过了很多组数据都是一样的结果,强行变正并不影响结果。
我的猜测是:因为数组中所有的数都是正数或者0,所以每一个up+down一定会大于0。
如果up小于0,那么它可以取到的所有的down都会使 up+down >= 0。
只是这些值是取过模的,所以会出现负数,我们需要将它放在0~(p-1)之间。