这是一个算法小白的起步……第一篇题解
前缀和: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 N N ( 1 ≤ N ≤ 50 , 000 1 \leq N \leq 50,000 1≤N≤50,000). 下 N N N
行包括 N N N 个整数值, 0 ≤ N i ≤ 1 , 000 , 000 0 \leq N_i \leq 1,000,000 0≤Ni≤1,000,000).
输出格式
输出符合要求的最长区间长度L
样例 #1
样例输入 #1
7
3
5
1
6
2
14
10
样例输出 #1
5
思路
猜个复杂度先:
5
×
1
0
4
5 \times 10^4
5×104,
O
(
n
3
)
O(n^3)
O(n3)必寄,因此暴力枚举起点和终点再加一起三循环寄中寄,也就没写暴力的code)
O
(
n
2
)
O(n^2)
O(n2)概率寄,试验后发现前十个不会T(来自小白の无聊试验)
大概率
O
(
1
)
−
O
(
n
2
)
不包括
O
(
n
2
)
O(1)-O(n^2)不包括O(n^2)
O(1)−O(n2)不包括O(n2)随机挑选一个幸运复杂度
然后
1
0
6
×
5
×
1
0
4
10^6 \times 5\times 10^4
106×5×104maybe要开个long long (后续好像用不到)
所以大概思路是用前缀和把复杂度压到
O
(
n
2
)
O(n^2)
O(n2),能有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;
}