萌新求助
查看原帖
萌新求助
419117
wangxiang123楼主2021/4/10 14:06

还没学前缀和,自己写的代码,为什么除了样例一个点都过不了QAQ,这个时间复杂度还能咋优化啊,原谅我看不太懂题解区。

我的思路是先遍历奇数点再遍历偶数点,求累加和。

#include<bits/stdc++.h>
using namespace std;
int n,m,col[100010],num[100010],ans;
int main()
{
	
	cin>>n>>m;
	for(int i=1;i<=n;i++)cin>>num[i];
	for(int i=1;i<=n;i++)cin>>col[i];
	for(int i=1;i<=n;i+=2)
	{
		for(int j=i+2;j<=n;j+=2)
		{
		if(col[i]==col[j])
        	ans+=(i+j)*(num[i]+num[j])%10007;
		}
	}
	for(int i=2;i<=n;i+=2)
	{
		for(int j=i+2;j<=n;j+=2)
		{
		if(col[i]==col[j])
        	ans+=(i+j)*(num[i]+num[j])%10007;
		}
	}
	cout<<ans;
	return 0;
}
```
2021/4/10 14:06
加载中...