Codeforces Round 909 (Div. 3)B

我的第一次代码只通过了第一个样例，第二个样例就超时了，完全暴力枚举。

#include<bits/stdc++.h>
using namespace std;
int t;
const int N=2e5+10;
typedef long long int ll;

ll a[N];

ll b[N];


ll ed=-1e6;
int main()
{
	memset(a,0,sizeof a);
	
	cin>>t;
	while(t--)
	{
		int n;
		cin>>n;
		for(int i=0;i<n;i++)
		{
			cin>>a[i]; 
//             scanf("%d",&a[i]);
		}
//		for(int i=0;i<n;i++)
//		{
//			cout<<a[i]<<" ";
//		}
//		cout<<"\n";
		ed=-1e6;
		for(int j=1;j<n;j++)
		{
			
			if(n%j==0)
			{
				memset(b, 0, sizeof b);
				ll m=0;
				
				for(int x=0;x<n/j;x++)
				{
					for(int p=0;p<j;p++)
					{
//						cout<<"m+p:"<<m+p<<" ";
						b[x]=b[x]+a[m+p];
//						cout<<"b"<<x<<":"<<b[x]<<" "; 
					}
//					cout<<"--------------"<<"\n";
					m=m+j;
				}
			}
//			for(int v=0;v<n/j;v++)
//			{
//				cout<<b[v]<<" "; 
//			}
//			cout<<"\n";
			sort(b,b+(n/j));
			
			ed=max( ed, b[n/j-1]-b[0]);
		}
		cout<<ed<<"\n";
	}	
	
	return 0;
} 

对于每个样例t都会有n的平方的时间复杂度。

之后看到别人的题解想到用前缀和来优化相加的过程

#include<bits/stdc++.h>
using namespace std;
int t,n;
long long a[200011],ans;
int main()
{
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n),ans=0;
		for(int i=1;i<=n;i++)scanf("%lld",&a[i]),a[i]+=a[i-1];
		for(int k=1;k<=n;k++)if(n%k==0)
		{
			long long mi=1e18,mx=0;
			for(int i=k;i<=n;i+=k)
			{
				mi=min(mi,a[i]-a[i-k]);
				mx=max(mx,a[i]-a[i-k]);
			}
			ans=max(ans,mx-mi);
		}
		printf("%lld\n",ans);
	}
	return 0;
}

这样时间复杂度的最大就变成了n*（n-k）

每隔k辆车，就进行一次比较，选出最大与最小。

最后输出答案ans。