题目1554：区间问题 map<int , vector<int> >的使用

最新推荐文章于 2025-04-23 01:29:20 发布

贫嘴小李子的幸福生活

最新推荐文章于 2025-04-23 01:29:20 发布

阅读量4.1k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： C++足迹

本文链接：https://blog.youkuaiyun.com/u013491262/article/details/21406351

C++足迹专栏收录该内容

274 篇文章

订阅专栏

本文探讨了在给定数组中寻找连续子数组的问题，该子数组的元素之和恰好等于给定的目标值k。通过引入前缀和数组和哈希表，实现了一种高效的搜索算法来解决这一经典计算机科学问题。

题目1554：区间问题

题目描述：

给定一个数组，判断数组内是否存在一个连续区间，使其和恰好等于给定整数k。

输入：

输入包含多组测试用例，每组测试用例由一个整数n(1<=n<=10000)开头，代表数组的大小。
接下去一行为n个整数，描述这个数组，整数绝对值不大于100。
最后一行为一个整数k(大小在int范围内)。

输出：

对于每组测试用例，若存在这个连续区间，输出其开始和结束的位置，s，e(s <= e)。
若存在多个符合条件的输出，则输出s较小的那个，若仍然存在多个，输出e较小的那个。
若不存在，直接输出"No"。

样例输入：

5
-1 2 3 -4 9
5
3
-1 2 -3
7
2
-1 1
0

样例输出：

2 3
No
1 2

记录前缀和 sum[i] = x[1] + x[2] + .... + x[i]

查找 x[s] + x[s+1] + ....+ s[e] = k ;

算法：

从小到大固定 s ,查找e 。

=> 固定s , 在sum[s+1] , sum[s+2] ,....sum[n]内查找sum[最小的e] = k+sum[s-1] 。

做映射: sum[i] -> i 。

注意sum[i]会有重复，这个时候做整数 -> 链表的映射。每次查找logN。

const int Max_N = 10008 ;
int x[Max_N] ;
int sum[Max_N] ;
int n  , k ;
map<int , vector<int> > st ;
 
int getint(){
    int t=0,flag=1;
    char c=getchar();
    while(c<'0'||c>'9'||c=='-'){
        if(c=='-')
           flag=-1 ;
        c=getchar();
    }
    while(c>='0'&&c<='9'){
       t=t*10+c-'0';
       c=getchar();
    }
   return t*flag;
}
 
 
 
void Ans(){
     int start , nowserch ;
     vector<int> now ;
     vector<int> ::iterator it ;
     for(start = 1 ; start <= n ; start++){
            nowserch = sum[start-1] + k ;
            if(st.find(nowserch) == st.end())
                continue ;
            now = st[nowserch] ;
            for(it = now.begin() ; it < now.end() ; it++){
                if((*it) >= start){
                    printf("%d %d\n" , start , (*it)) ;
                    return ;
                }
            }
     }
     puts("No") ;
     return ;
}
 
int  main(){
     int i ;
     while(scanf("%d" ,&n) != EOF){
          st.clear() ;
          sum[0] = 0 ;
          for(i = 1 ; i <= n ; i++){
             x[i] = getint() ;
             sum[i] = sum[i-1] + x[i] ;
             st[sum[i]].push_back(i) ;
          }
          scanf("%d" ,&k) ;
          Ans() ;
     }
     return  0 ;
}