中二羊专题:栋栋的入门题(前缀和)

原题

标题虽然是栋栋的入门题,但它并不是入门题。
原题的题目描述是:

给出N个整数,以及M个求和范围,求出每一个范围的数字的和。

提示:显然,这并不是一道入门题。

这就要用到一种新的思想:前缀思想。

进入正题

数组 \(a\) (此处方便讲解,忽略下标 \(0\) )有这 \(5\) 个数字: \(1,3,2,1,5\) 。你求 \(a_1+...+a_3\) 是以下代码:

for(int i=1;i<=3;i++)ans+=a[i];

再假设数组 \(b\)\(m\) 个数字。你需要求 \(b_x+...+b_y\) 。一般人用以下代码:

for(int i=x;i<=y;i++)ans+=b[i];

你有 \(n\) 次访问(即求 \(n\) 次从 \(b_x+...+b_y\) )。那么你代码应该会是这样:

for(int i=1;i<=n;i++){
	cin>>x>>y;
	for(int j=x;j<=y;j++)ans+=b[j];
}

但这样子太慢了。
但是我们还有一种方法解决。
我们可以令 \(c_i\)\(b_1+b_2+...+b_i\) 。那么求 \(b_x+...+b_y\) 就非常快了。 \(c_y-c_{x-1}\)

解释

已知 \(c_y=b_1+...+b_y\)\(c_x=b_1+...+b_x\) 。所以说(此处化简):
\(c_y-c_{x-1}=b_1+...+b_y-(b_1+...+b_{x-1})=b_x+...+b_y\)

代码

建议别抄。这里只是给一个例子。

#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<cctype>
namespace FastIo{
    typedef unsigned long long ull;
    typedef __uint128_t L;
    static char buf[100000],*p1=buf,*p2=buf;
    #define gc p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++
    class FastMod{
        public:
            FastMod(ull b):b(b),m(ull((L(1)<<64)/b)){}
            ull reduce(ull a){
                ull q=(ull)((L(m)*a)>>64);
                ull r=a-q*b;
                return r>=b?r-b:r;
            }
        private:
            ull b,m;
    };
    FastMod QM(10);
    class QIO{
    	public:
    		inline void read(int &x){
    			x=0,ch=gc;
    			while(!isdigit(ch))ch=gc;
    			while(isdigit(ch)){x=(x<<3)+(x<<1)+(ch^48);ch=gc;}
			}
			void write(int a){
				if(a>9)write(a/10);
				putchar(QM.reduce(a)^48);
			}
		private:
			char ch;
	};
	QIO qrw;
}
using namespace FastIo;
int a[1000001];
signed main(){
	register int i(1);
	int n,m,x,y;
	qrw.read(n);
	qrw.read(m);
	for(;i<=n;i=-~i){
		qrw.read(x);
		a[i]=a[i-1]+x;
	}
	for(i=1;i<=m;i=-~i){
		qrw.read(x);
		qrw.read(y);
		qrw.write(a[y]-a[x-1]);
		putchar('\n');
	}
    exit(0);
    return 0;
}

后记: \(N\) 年前的东西了,看个乐子就行了。

posted @ 2023-05-14 10:58  SHOJYS  阅读(14)  评论(0编辑  收藏  举报