ny 108 士兵杀敌(一)-- 水(大家都是这么说的)利用线段树果断超时。。。

来源:互联网 发布:win10电脑优化 编辑:程序博客网 时间:2024/04/29 01:39

士兵杀敌(一)

时间限制:1000 ms  |  内存限制:65535 KB
难度:3
描述

南将军手下有N个士兵,分别编号1到N,这些士兵的杀敌数都是已知的。

小工是南将军手下的军师,南将军现在想知道第m号到第n号士兵的总杀敌数,请你帮助小工来回答南将军吧。

注意,南将军可能会问很多次问题。

输入
只有一组测试数据
第一行是两个整数N,M,其中N表示士兵的个数(1<N<1000000),M表示南将军询问的次数(1<M<100000)
随后的一行是N个整数,ai表示第i号士兵杀敌数目。(0<=ai<=100)
随后的M行每行有两个整数m,n,表示南将军想知道第m号到第n号士兵的总杀敌数(1<=m,n<=N)。
输出
对于每一个询问,输出总杀敌数
每个输出占一行
样例输入
5 21 2 3 4 51 32 4
样例输出
69


利用线段树果断超时,以下是超时代码

#include <stdio.h>#include <string.h>#include <algorithm>using namespace std;const int N = 1E6+5;struct tree{    int l,r;    int num;}t[4*N];void build(int l,int r,int root){    t[root].l=l;    t[root].r=r;    t[root].num=0;    if(l==r){        return;    }    int mid=(l+r)/2;    build(l,mid,2*root);    build(mid+1,r,2*root+1);}void update(int val,int id,int root){    if(t[root].l==t[root].r&&t[root].l==id){        t[root].num+=val;        return;    }    int mid=(t[root].l+t[root].r)/2;    if(id<=mid)        update(val,id,2*root);    else        update(val,id,2*root+1);    t[root].num=t[2*root].num+t[2*root+1].num;}int ans;void find(int l,int r,int root){    if(t[root].l==l&&t[root].r==r){        ans+=t[root].num;        return;    }    int mid=(t[root].l+t[root].r)/2;    if(r<=mid)        find(l,r,2*root);    else if(l>mid)        find(l,r,2*root+1);    else{        find(l,mid,2*root);        find(mid+1,r,2*root+1);    }}int main(){    int t,n,i,j;    while(~scanf("%d %d",&n,&t)){        build(1,n,1);        for(i=1;i<=n;i++){            int x;            scanf("%d",&x);            update(x,i,1);        }        while(t--){            int x,y;            scanf("%d %d",&x,&y);ans=0;find(x,y,1);printf("%d\n",ans);        }    }    return 0;}


可以设置两个数组,a[i] 存第i个士兵杀敌的数目,b[i] 存前i个士兵杀敌的数目

那么从x到y个士兵杀敌的数目是:b[y]-b[x-1]

#include <stdio.h>#include <string.h>#include <algorithm>using namespace std;const int N = 1E6+5;int a[N];int b[N];int main(){int n,m,x,y,i;scanf("%d %d",&n,&m);memset(a,0,sizeof(a));memset(b,0,sizeof(b));for(i=1;i<=n;i++){scanf("%d",&a[i]);b[i]=b[i-1]+a[i];}while(m--){scanf("%d %d",&x,&y);printf("%d\n",b[y]-b[x-1]);}return 0;}



0 0
原创粉丝点击