码迷,mamicode.com
首页 > 其他好文 > 详细

[mock]8月8日

时间:2014-08-09 00:00:26      阅读:352      评论:0      收藏:0      [点我收藏+]

标签:blog   for   问题   div   代码   line   amp   log   

第一题是整数的方阵,求其中的子方阵,和最大。返回最大和以及子方阵宽度。因为做了topcoder的题,所以比较顺手,O(n^3)的复杂度。

pair<int,int> maxiSum(vector<vector<int> > &a) {  //first is n second is sum
    int N = a.size();
    int retVal = INT_MIN;
    int n = 1;
    // fix two columns, i, j
    for (int i = 0; i < N; i++) {
        vector<int> sum(N);
        for (int j = i; j < N; j++) {
            // calc the sum per line
            for (int k = 0; k < N; k++) {
                    sum[k] = sum[k] + a[k][j];
            }
            // calc the sum of the squere with i, j
            int tmpSum = 0;
            int w = j - i;
            for (int m = 0; m < N; m++) {
                if (m < w) {
                    tmpSum += sum[m];
                } else if (m == w) {
                    tmpSum += sum[m];
                    if (tmpSum > retVal) {
                        retVal = tmpSum;
                        n = w + 1;
                    }
                } else {
                    tmpSum += sum[m];
                    tmpSum -= sum[m - w - 1];
                    if (tmpSum > retVal) {
                        retVal = tmpSum;
                        n = w + 1;
                    }
                }
            }
        }
    }
    return make_pair(n, retVal);
}

第二题,是有个0和1的矩阵,求里面每个位置到最近的0的距离(类似曼哈顿距离)。这个题一开始想动态规划什么的,都不合适,比较像洪泛,但如果从每个0或1出发,就效率很低,n^4和完全枚举是一样的。后来想到和之前topcoder的一道题挺像,可以先把0全部放到queue里面,然后逐步往外扩,便可求得。

void make(vector<vector<int> > &matrix) {
    int N = matrix.size();
    vector<vector<int> > dist;
    vector<vector<bool> > visited;
    dist.resize(N);
    visited.resize(N);
    for (int i = 0; i < N; i++) {
        dist[i].resize(N);
        visited[i].resize(N);
    }
    queue<pair<int, int> > que;
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            if (matrix[i][j] == 0) {
                dist[i][j] = 0;
                visited[i][j] = true;
                que.push(make_pair(i, j));
            }
        }
    }
    while (!que.empty()) {
        pair<int, int> pp = que.front();
        que.pop();
        
        if (valid(pp.first - 1, pp.second, visited)) {
            dist[pp.first - 1][pp.second] = dist[pp.first][pp.second] + 1;
            que.push(make_pair(pp.first - 1, pp.second));
            visited[][] = true;
        }
        if (valid(pp.first, pp.second - 1, visited)) {
        
        }
        if (valid(pp.first + 1, pp.second, visited)) {
        
        }
        if (valid(pp.first, pp.second + 1, visited)) {
        
        }
    }
    
    // copy to matrix
    matrix = dist;
}

bool valid(int i, int j, vector<vector<bool> > & visited) {
    // return whether we need to process this elem
    int N = visited.size();
    if (i < 0 || j < 0 || i >= N || j >= N || visited[i][j]) {
        return false;
    }
    return true;
}

遍历四个方向的代码其实应该更精简,类似如下:

dx[] = {-1,0,1,0}
dy[] = {0,1,0,-1}

x + dx[i]
y + dy[i]

第三题当时没有想出来,是求一个数组中子段和的绝对值最小值。比如如下数组,就是1+2+ -3的绝对值最小。

{1,2,-3,-100,3}

这个题目很自然会想到去用最大子段和的做法,但是(应该)不能类似动态规划降到O(n)。关键是,不满足最优子问题的特性。比如该数组中前三个元素的最优解是0,前两个元素的最优解是2,0没有用到之前最优解地方。

那么正确的做法是可以记录之前的前缀和,然后排序,就可以求得了。因为:|sum[i..j]| = |p[j] - p[i - 1]|,p是前缀和,那么排完序后,相邻的两个就能求得绝对值最小,只是要注意,i从0开始,那么p[-1]就是不包括0的情况,就是0,要在前缀和数组里加一个0.

本题的前缀和数组排序完了就是如下,里面两个0,一个开始之前p[-1],一个p[2],就能求出sum[0..2]=0,绝对值最小。

{-97, -94,  0, 0, 1, 3}

  

[mock]8月8日,布布扣,bubuko.com

[mock]8月8日

标签:blog   for   问题   div   代码   line   amp   log   

原文地址:http://www.cnblogs.com/lautsie/p/3900303.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!