zoukankan      html  css  js  c++  java
  • 51 nod 1624 取余最长路 思路:前缀和 + STL(set)二分查找

    题目:

    写这题花了我一上午时间。

    下面是本人(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)之间。

  • 相关阅读:
    Js
    CSS
    CSS
    第七周作业及总结
    第六周作业及总结
    第五周作业及总结
    第四周作业及总结
    第三周作业及总结
    7-1 判断上三角矩阵
    第二周作业及总结
  • 原文地址:https://www.cnblogs.com/zhangjiuding/p/7424083.html
Copyright © 2011-2022 走看看