[NOIP2015]求和

第一行是用一个空格隔开的两个正整数 n 和 m,n 表纸带上格子的个数， m 表纸带上颜色的种类数。
第二行有 n 用空格隔开的正整数，第 i 数字 number 表纸带上编号为 i 格子上面写的数字。
第三行有 n 用空格隔开的正整数，第 i 数字 color 表纸带上编号为 i 格子染的颜色。

C++14(g++5.4) 解法, 执行用时: 68ms, 内存消耗: 1380K, 提交时间: 2019-11-09 19:48:11

#include<bits/stdc++.h>
#pragma loop_opt(on)
using namespace std;

int a[100001],c[100001],s[3][100001],x[3][100001];

int main()
{
	int n,m,i,j;
	long long ans=0;
	cin>>n>>m;
	for(i=1;i<=n;i++)cin>>a[i];
	for(i=1;i<=n;i++){
		cin>>c[i];
		s[i%2][c[i]]=(s[i%2][c[i]]+a[i])%10007;
		x[i%2][c[i]]++;
	}
	for(i=1;i<=n;i++)ans=(ans+i*(s[i%2][c[i]]%10007+(x[i%2][c[i]]-2)*a[i]%10007))%10007;
	cout<<ans%10007;
	return 0;
}

C++(g++ 7.5.0) 解法, 执行用时: 64ms, 内存消耗: 2348K, 提交时间: 2023-04-18 21:57:42

#include<bits/stdc++.h>
using namespace std;
int s[100001][2],x[100001],c[100001],sum[100001][2];
int main()
{
	int n,m,ans=0;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	cin>>x[i];
	for(int i=1;i<=n;i++)
	{
		cin>>c[i];
		s[c[i]][i%2]++;
		sum[c[i]][i%2]=(sum[c[i]][i%2]+x[i])%10007;
	}
	for(int i=1;i<=n;i++)
	ans=(ans+i*((s[c[i]][i%2]-2)*x[i]%10007+sum[c[i]][i%2]))%10007;
	cout<<ans<<"\n";
	return 0;
}

C++(clang++ 11.0.1) 解法, 执行用时: 54ms, 内存消耗: 1264K, 提交时间: 2023-07-17 12:42:58

#include<iostream>
using namespace std;
int s[100001][2],x[100001],c[100001],sum[100001][2];
int n;
#define for for(int i=1;i<=n;i++)
int main()
{
	int n,m,ans=0;
	cin>>n>>m;
	for cin>>x[i];
	for{
		cin>>c[i];
		s[c[i]][i%2]++;
		sum[c[i]][i%2]=(sum[c[i]][i%2]+x[i])%10007;
	}
	for ans=(ans+i*((s[c[i]][i%2]-2)*x[i]%10007+sum[c[i]][i%2]))%10007;
	cout<<ans<<endl;
	return 0;
}

详情