前缀和：P3131 [USACO16JAN] Subsequences Summing to Sevens S

Johncronus

已于 2024-02-28 13:47:53 修改

阅读量1k

点赞数 12

分类专栏： # P2.8前缀和和差分文章标签： c++

于 2024-02-28 13:43:54 首次发布

本文链接：https://blog.csdn.net/Johncronus/article/details/136343847

版权

P2.8前缀和和差分专栏收录该内容

4 篇文章 0 订阅

订阅专栏

文章讲述了如何使用前缀和优化算法解决一个关于子序列和被7整除的问题，展示了从暴力枚举到O(n^2)再到潜在O(1)的时间复杂度提升过程。

摘要由CSDN通过智能技术生成

这是一个算法小白的起步……第一篇题解

前缀和：P3131 [USACO16JAN] Subsequences Summing to Sevens S

题目来自洛谷。

[USACO16JAN] Subsequences Summing to Sevens S

题目描述

给N个数，分别是a[1],a[2],…,a[n]。求一个最长的区间[x,y]，使得区间中的数(a[x],a[x+1],a[x+2],…,a[y-1],a[y])的和能被7整除。输出区间长度。若没有符合要求的区间，输出0。

输入格式

第一行为 $N$ ( $\leq N \leq 50,000$ ). 下 $N$

行包括 $N$ 个整数值， $\leq N_i \leq 1,000,000$ ).

输出格式

输出符合要求的最长区间长度L

样例 #1

样例输入 #1

样例输出 #1

思路

猜个复杂度先:
$\times 10^4$ , $O(n^3)$ 必寄,因此暴力枚举起点和终点再加一起三循环寄中寄，也就没写暴力的code）
$O(n^2)$ 概率寄，~~试验后发现前十个不会T（来自小白の无聊试验）~~
大概率 $O(1)-O(n^2)不包括O(n^2)$ 随机挑选一个幸运复杂度

然后 $10^6 \times 5\times 10^4$ maybe~~要开个long long~~ (后续好像用不到)

所以大概思路是用前缀和把复杂度压到 $O(n^2)$ ，能有50 70 左右？
贴个code

	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	int n;
	cin>>n;
	memset(ad, 0, sizeof ad);
	for(int i=1; i<=n; ++i){
		long long k;
		cin>>k;
		ad[i]=ad[i-1]+k;
	}
	int len=0;
	for(int i=1; i<=n; ++i){
		for(int j=i; j<=n; ++j){
			if((ad[j]-ad[i-1])%7==0)len=max(len, j-i+1);
		}
	}
	cout<<len;

注意到是mod 7==0的长度，所以可以不去暴力枚举起点和终点，只需要找到两个%7相等的端点，这样相减后所得区间和一定可以被7整除。
压完的话我是O(n)解决的……~~但我在洛谷看到了惊人的O(1)字样（但是我没细看，我不信）~~

#include<iostream>
#include<cstring>
using namespace std;
const int N=110000;
long long ad[N];
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	int n;
	cin>>n;
	memset(ad, 0, sizeof ad);
	for(int i=1; i<=n; ++i){
		long long k;
		cin>>k;
		ad[i]=ad[i-1]+k;
	}
	
	for(int i=1; i<=n; ++i)ad[i]%=7;
	int maxl=0;
	for(int i=0; i<=6; ++i){
		int st=0, en=n;
		while(ad[st]!=i){
			st++;
			if(st>n)break;
		}
		while(ad[en]!=i){
			en--;
			if(en<0)break;
		}
		if(en<=st)continue;
		maxl=max(en-st, maxl);
	}
	cout<<maxl;
	return 0;
}