51Nod 1094 和为k的连续性 c/c++题解

题目描述

一整数数列a1, a2, … , an（有正有负），以及另一个整数k，求一个区间[i, j]，(1 <= i <= j <= n)，使得a[i] + … + a[j] = k。
输入
第1行：2个数N,K。N为数列的长度。K为需要求的和。(2 <= N <= 10000，-10^9 <= K <= 10^9)
第2 - N + 1行：A[i](-10^9 <= A[i] <= 10^9)。
输出
如果没有这样的序列输出No Solution。
输出2个数i, j，分别是区间的起始和结束位置。如果存在多个，输出i最小的。如果i相等，输出j最小的。
输入样例
6 10
1
2
3
4
5
6
输出样例
1 4

题解：

本来考察的只是把O(n³)的时间复杂度优化到O(n²)，也就是第一种方法，采用前缀和，但是为了巩固下树状数组和线段树的区间求和知识，我就实现了三种方法，具体看代码，如果这两种数据结构不理解，请看下别人的博客(以后我也可能会自己写博客，敬请期待)。

• 第一种方法：前缀和的优化，很简单，不想多说了 — O(n²)
• 第二种方法：树状数组，区间求和的模板套用 — O(n²log₂n)
• 第三种方法：线段树，区间求和的模板套用 — O(n²log₂n)
  注意：
  这道题目用线段树的方法并不能通过，也就是会TLE，对比一下树状数组，虽然区间查询的时间复杂度都是O(log₂n)的，但是线段树的常数要大，如果只是一般的题目(不卡常数)，两种方法一定都能过。
  反正能用数组数组就别用线段树，时间相对更优而且更好写，但是线段树的应用范围更广。

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <cstring>
#include <string>
#include <algorithm>
#include <vector>
#include <deque>
#include <list>
#include <utility>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <bitset>
#include <iterator>
using namespace std;

typedef long long ll;
const int inf = 0x3f3f3f3f;
const ll  INF = 0x3f3f3f3f3f3f3f3f;
const int PI = acos(-1);
const int E = exp(1);
const int MOD = 1e9+7;
const int MAX = 1e4+5;
int n,k;
int a[MAX]; 

int tree[MAX];// 树状数组
int lowbit(int x)
{
    return x&(-x);
}
void InitTree()
{
    for(int i = 1; i <= n; i++)
    {
        for(int j = i - lowbit(i)+1; j <= i; j++)
        {
            tree[i] += a[j];
        }
    }
}
int AskRangeSum1(int a,int b)
{
    int sum1 = 0;
    int sum2 = 0;
    for(int i = a-1; i > 0; i -= lowbit(i))
    {
        sum1 += tree[i];
    }
    for(int i = b; i > 0; i -= lowbit(i))
    {
        sum2 += tree[i];
    }
    return sum2 - sum1;
}

struct SegTree
{
    int sum,left,right;
    int lazy_tag;
}segTree[4*MAX];
void build(int root,int left,int right)
{
    if(left == right)
    {
        int value;
        cin >> value;
        segTree[root].sum = value;
    }
    else
    {
        int mid = (left + right) / 2;
        build(root*2,left,mid);
        build(root*2+1,mid+1,right);
        segTree[root].sum = segTree[root*2].sum + segTree[root*2+1].sum;
        segTree[root].left = left;
        segTree[root].right = right;
    }
}
void PushDown(int root)
{
    if(segTree[root].lazy_tag)
    {
        /** 1.将根节点的懒标记下传到左右子节点 */
        segTree[root*2].lazy_tag += segTree[root].lazy_tag;
        segTree[root*2+1].lazy_tag += segTree[root].lazy_tag;

        /** 2.左右子节点更新区间和 */
        segTree[root*2].sum += segTree[root].lazy_tag * (segTree[root*2].right - segTree[root*2].left);
        segTree[root*2+1].sum += segTree[root].lazy_tag * (segTree[root*2].right - segTree[root*2].left);

        /** 3.根节点懒标记置为0 */
        segTree[root].lazy_tag = 0;
    }
}
int AskRangeSum2(int a,int b,int root,int left,int right)
{
    /** 查询给定区间[a,b]的和 */
    if(b < left || a > right)
    {
        return 0;
    }
    if(a <= left && b >= right)
    {
        /** 当前区间[left,right]在查询区间[a,b]内，直接返回当前区间的和 */
        return segTree[root].sum;
    }
    else
    {
        PushDown(root);
        int sum1,sum2;
        int mid = (left + right) / 2;
        sum1 = AskRangeSum2(a,b,root*2,left,mid);
        sum2 = AskRangeSum2(a,b,root*2+1,mid+1,right);
        return sum1+sum2;
    }
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    while(cin >> n >> k)
    {
    	/* 第一种方法：前缀和优化法 --- O(n^2) */
        for(int i = 1; i <= n; i++)
        {
            cin >> a[i];
            a[i] += a[i-1];
        }
        bool flag = false;
        int s,e;
        for(int i = 1; i <= n; i++)
        {
            for(int j = i; j <= n; j++)
            {
                if(a[j] - a[i-1] == k)
                {
                    s = i; e = j;
                    flag = true;
                    break;
                }
            }
            if(flag) break;
        }
        if(flag)
            cout << s << " " << e << endl;
        else
            cout << "No Solution" << endl;*/

        /* 第二种方法：树状数组法 --- O(n^2logn)
        for(int i = 1; i <= n; i++)
        {
            cin >> a[i];
        }
        InitTree();
        bool flag = false;
        int s,e;
        for(int i = 1; i <= n; i++)
        {
            for(int j = i; j <= n; j++)
            {
                if(AskRangeSum1(i,j) == k)
                {
                    s = i; e = j;
                    flag = true;
                }
            }
            if(flag) break;
        }
        if(flag)
            cout << s << " " << e << endl;
        else
            cout << "No Solution" << endl;
        */

        /* 第三种方法：线段树法 --- O(n^2logn) 
        build(1,0,n-1);
        bool flag = false;
        int s,e;
        for(int i = 1; i <= n; i++)
        {
            for(int j = i; j <= n; j++)
            {
                if(AskRangeSum2(i,j,1,1,n) == k)
                {
                    s = i; e = j;
                    flag = true;
                }
            }
            if(flag) break;
        }
        if(flag)
            cout << s << " " << e << endl;
        else
            cout << "No Solution" << endl;
    }
    */
    return 0;
}