还没学前缀和,自己写的代码,为什么除了样例一个点都过不了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;
}
```