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 <bits\stdc++.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)之间。

时间: 2024-10-08 11:43:10

51 nod 1624 取余最长路 思路:前缀和 + STL(set)二分查找的相关文章

1624 取余最长路(set)

1624 取余最长路 基准时间限制:1 秒 空间限制:131072 KB 分值: 40 难度:4级算法题 佳佳有一个n*m的带权矩阵,她想从(1,1)出发走到(n,m)且只能往右往下移动,她能得到的娱乐值为所经过的位置的权的总和. 有一天,她被下了恶毒的诅咒,这个诅咒的作用是将她的娱乐值变为对p取模后的值,这让佳佳十分的不开心,因为她无法找到一条能使她得到最大娱乐值的路径了! 她发现这个问题实在是太困难了,既然这样,那就只在3*n的矩阵内进行游戏吧! 现在的问题是,在一个3*n的带权矩阵中,从(

1624 取余最长路 (前缀和+set二分)

1624 取余最长路 佳佳有一个n*m的带权矩阵,她想从(1,1)出发走到(n,m)且只能往右往下移动,她能得到的娱乐值为所经过的位置的权的总和. 有一天,她被下了恶毒的诅咒,这个诅咒的作用是将她的娱乐值变为对p取模后的值,这让佳佳十分的不开心,因为她无法找到一条能使她得到最大娱乐值的路径了! 她发现这个问题实在是太困难了,既然这样,那就只在3*n的矩阵内进行游戏吧! 现在的问题是,在一个3*n的带权矩阵中,从(1,1)走到(3,n),只能往右往下移动,问在模p意义下的移动过程中的权总和最大是多

51nod1624 取余最长路 前缀和 + set

由于只有3行,因此只会会换行2次,假设$x, y$分别为这两次的换行点 那么答案为$S[1][x] +S[2][y] - S[2][x - 1] + S[3][n] - S[3][y - 1]$ 其中,$S[i]$表示第$i$行的前缀和 令$a[x] = S[1][x] - S[2][x - 1], b[y] = S[2][y] - S[3][y - 1]$ 考虑枚举$x$,那么问题转化为询问在一堆数中求一个数$k$使得$v (= a[x] + S[3][n]) + k \;mod\;p$最大

51nod 1624 取余最短路(set)

题意: 佳佳有一个n*m的带权矩阵,她想从(1,1)出发走到(n,m)且只能往右往下移动,她能得到的娱乐值为所经过的位置的权的总和. 有一天,她被下了恶毒的诅咒,这个诅咒的作用是将她的娱乐值变为对p取模后的值,这让佳佳十分的不开心,因为她无法找到一条能使她得到最大娱乐值的路径了! 她发现这个问题实在是太困难了,既然这样,那就只在3*n的矩阵内进行游戏吧! 现在的问题是,在一个3*n的带权矩阵中,从(1,1)走到(3,n),只能往右往下移动,问在模p意义下的移动过程中的权总和最大是多少. 实际上路

51 nod 1097 拼成最小的数 思路:字符串排序

题目: 思路:1.以字符串输入这些整数. 2.对这些字符串排序,排序规则为尽量让能让结果变小的靠前. 代码中有注释,不懂的欢迎在博客中评论问我. 代码: #include <bits\stdc++.h> using namespace std; string a[10001]; //比较规则:尽量让结果字符串最小 bool cmp(string a,string b){ return a+b <= b+a; } int main(){ int n; cin >> n; for

51 nod 1055 最长等差数列(dp)

1055 最长等差数列 基准时间限制:2 秒 空间限制:262144 KB 分值: 80 难度:5级算法题 N个不同的正整数,找出由这些数组成的最长的等差数列. 例如:1 3 5 6 8 9 10 12 13 14 等差子数列包括(仅包括两项的不列举) 1 3 5 1 5 9 13 3 6 9 12 3 8 13 5 9 13 6 8 10 12 14 其中6 8 10 12 14最长,长度为5. Input 第1行:N,N为正整数的数量(3 <= N <= 10000). 第2 - N+1行

poj 3349:Snowflake Snow Snowflakes(哈希查找,求和取余法+拉链法)

Snowflake Snow Snowflakes Time Limit: 4000MS   Memory Limit: 65536K Total Submissions: 30529   Accepted: 8033 Description You may have heard that no two snowflakes are alike. Your task is to write a program to determine whether this is really true. Y

51 nod 1427 文明 (并查集 + 树的直径)

1427 文明 题目来源: CodeForces 基准时间限制:1.5 秒 空间限制:131072 KB 分值: 160 难度:6级算法题 安德鲁在玩一个叫“文明”的游戏.大妈正在帮助他. 这个游戏里面有n个城市和m条双向的道路.城市从1到n编号.对于每一对城市,他们之间要么有唯一的一条道路,要么就是不可互达.一条道路的定义是一个包含不同城市的序列 v1, v2,...,vk ,  vi  和  vi+1 (1≤ i < k)之间有直接的一条道路相连.这条道路的长度是k-1.两个城市在同一区域的

P1807 最长路_NOI导刊2010提高(07)

P1807 最长路_NOI导刊2010提高(07) 题目描述 设G为有n个顶点的有向无环图,G中各顶点的编号为1到n,且当为G中的一条边时有i < j.设w(i,j)为边的长度,请设计算法,计算图G中<1,n>间的最长路径. 输入输出格式 输入格式: 输入文件longest.in的第一行有两个整数n和m,表示有n个顶点和m条边,接下来m行中每行输入3个整数a,b,v(表示从a点到b点有条边,边的长度为v). 输出格式: 输出文件longest.out,一个整数,即1到n之间的最长路径.如