九度oj 1554 区间问题

原题链接：http://ac.jobdu.com/problem.php?pid=1554

由数列的前缀和：$\begin{align*}\Large{} S_n &=\Large{}\sum_{i=1}^{n}{{a_i}} \ \ \ \ i=1,2,3...n\end{align*}$

由于：$\begin{align*}\Large{} S_n -S_{n-1}&=\Large{}a_n \end{align*}$

所以区间$\begin{align*}\Large{} [i,j]\end{align*}$ 之和，$\begin{align*}\Large{} S_j -S_{i-1}&=\Large{}a_i+a_{i+1}+...+a_j\ \ \ 1 \leq i,j,\leq n\end{align*}$

由题意给定一个数组，判断数组内是否存在一个连续区间，使其和恰好等于给定整$\begin{align*}\Large{} k \end{align*}$

其实就是判断$\begin{align*}\Large{} Sum[j] -Sum[i]=k\end{align*}$是否成立。

暴力的方法直接枚举，由于$\begin{align*}\Large{} n=10000\end{align*}$会$\begin{align*}tle\end{align*}$。

现在我们换个思路，试试二分，考虑开个结构体数组，保存原数组的前缀和+原来的位置。

然后对其排序，再进行二分。但这样可能会出错，所以再判断一下，即可。

懒得啰嗦了，直接看代码吧。。

 #include<algorithm>
 #include<iostream>
 #include<cstdlib>
 #include<cstring>
 #include<cstdio>
 using std::sort;
 using std::lower_bound;
 const int Max_N = ;
 int arr[Max_N], sum[Max_N], temp[Max_N];
 struct Node {
     int v, pos;
     bool operator<(const Node &x) const {
         if (v == x.v) return pos < x.pos;
         return v < x.v;
     }
 }rec[Max_N];
 void solve(int n, int k) {
     int p = ;
     for (int i = ; i < n; i++) {
         p = lower_bound(temp + , temp + n, sum[i] + k) - temp;
         if (p == n && sum[n] != k) continue;
         if (rec[p].pos > i && rec[p].v == sum[i] + k) {
             printf("%d %d\n", i + , rec[p].pos);
             return;
         } else {
             p++;
             while (rec[p].v == sum[i] + k) {
                 if (rec[p].pos > i) {
                     printf("%d %d\n", i + , rec[p].pos);
                     return;
                 }
                 p++;
             }
         }
     }
     puts("No");
 }
 int main() {
 #ifdef LOCAL
     freopen("in.txt", "r", stdin);
     freopen("out.txt", "w+", stdout);
 #endif
     int n, k;
     while (~scanf("%d", &n)) {
         for (int i = ; i <= n; i++) {
             scanf("%d", &arr[i]);
             sum[i] = sum[i - ] + arr[i];
             rec[i].v = sum[i], rec[i].pos = i;
         }
         scanf("%d", &k);
         sort(rec + , rec + n);
         for (int i = ; i <= n; i++) temp[i] = rec[i].v;
         solve(n, k);
     }
     return ;
 }