牛客竞赛——求和（简单排序）

 

 我的排序+暴力：

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<vector>
#include<cmath>
#include<string>
#include<map>
#include<queue>
using namespace std;
typedef long long ll;
struct node{
	ll id,color,number;
	ll multi;
}a[100010];
int flag[100010];
const ll mod=10007;
bool cmp(node x,node y){
	if(x.color==y.color)
		return x.id<y.id;
	return x.color<y.color;
}
int main(){
	ll n,m;
	scanf("%lld %lld",&n,&m);
	for(int i=0;i<n;i++){
		scanf("%lld",&a[i].number);
		a[i].id=i+1;
		a[i].multi=a[i].id*a[i].number;//先求自身的乘积 
	}
	for(int i=0;i<n;i++){
		scanf("%lld",&a[i].color);
		flag[a[i].color]++;//记录颜色出现次数 
	}
	sort(a,a+n,cmp);
	ll cnt[2]={0,0};//cnt[0]某种颜色对应的奇数下标的个数.... 
	ll cnt_1[2]={0,0};//cnt_1[0]某种颜色奇数下标与自身number乘积的前缀和.... 
	ll cnt2[2]={0,0};//cnt2[0]某种颜色奇数下标前缀和，cnt2[1]奇数下标对应的number前缀和 
	ll cnt3[2]={0,0};//cnt3[0]某种颜色偶数下标前缀和，cnt3[1]偶数下标对应的number前缀和
	ll sum=0;
	for(int i=0;i<n;i+=flag[a[i].color]){
		if(flag[a[i].color]<=1)continue;
		for(int j=i;j<i+flag[a[i].color];j++){
			if(a[j].id&1){
				cnt_1[0]+=a[j].multi;
				cnt[0]++;
				cnt2[0]+=a[j].id;
				cnt2[1]+=a[j].number;
			}
			else {
				cnt_1[1]+=a[j].multi;
				cnt[1]++;
				cnt3[0]+=a[j].id;
				cnt3[1]+=a[j].number;
			}
		}
		//奇(偶)数下标对都至少2个块 
		if(cnt[0]>1){
			//推导公式 
			sum=(sum+(cnt[0]-2)%mod*cnt_1[0]%mod+cnt2[0]%mod*cnt2[1]%mod)%mod;
		}
		if(cnt[1]>1){
			sum=(sum+(cnt[1]-2)%mod*cnt_1[1]%mod+cnt3[0]%mod*cnt3[1]%mod)%mod;
		}
		cnt[0]=cnt[1]=0;
		cnt_1[0]=cnt_1[1]=0;
		cnt2[0]=cnt2[1]=0;
		cnt3[0]=cnt3[1]=0;
	}
	cout<<sum%mod<<endl;
	return 0;
}
//            /\       |  /  |**、
//			 /  \      | /   |   \
//			/    \     |/    |   /  _____                      ____   |  /
//		   /------\    |\    |__/  /     \  \      /\      /  /    \  | /
//		  /        \   | \   |    /       \  \    /  \    /  /______\ |/
//		 /          \  |  \  |    \       /   \  /    \  /   \        |
//      /            \ |   \ |     \_____/     \/      \/     \_____  |
/**
 *　　　　　　　　┏┓　　 　┏┓
 * 　　　　　　　┏┛┗━━━━━━━┛┗━━━┓
 * 　　　　　　　┃　　　　　　　┃ 　
 * 　　　　　　　┃　　　━　　 　┃
 * 　　　　　　　┃　＞　　　＜　┃
 * 　　　　　　　┃　　　　　　　┃
 * 　　　　　　　┃...　⌒　... 　┃
 * 　　　　　　　┃　　　　　　　┃
 * 　　　　　　　┗━┓　　　┏━┛
 * 　　　　　　　　　┃　　　┃　Code is far away from bug with the animal protecting　　　　　　　　　　
 * 　　　　　　　　　┃　　　┃   神兽保佑,代码无bug
 * 　　　　　　　　　┃　　　┃　　　　　　　　　　　
 * 　　　　　　　　　┃　　　┃  　　　　　　
 * 　　　　　　　　　┃　　　┃
 * 　　　　　　　　　┃　　　┃　　　　　　　　　　　
 * 　　　　　　　　　┃　　　┗━━━┓
 * 　　　　　　　　　┃　　　　　　　┣┓
 * 　　　　　　　　　┃　　　　　　　┏┛
 * 　　　　　　　　　┗┓┓┏━┳┓┏┛
 * 　　　　　　　　　　┃┫┫　┃┫┫
 * 　　　　　　　　　　┗┻┛　┗┻┛
 */
// warm heart, wagging tail,and a smile just for you!
//
//                            _ooOoo_
//                           o8888888o
//                           88" . "88
//                           (| -_- |)
//                           O\  =  /O
//                        ____/`---'\____
//                      .'  \|     |//  `.
//                     /  \|||  :  |||//  \
//                    /  _||||| -:- |||||-  \
//                    |   | \\  -  /// |   |
//                    | \_|  ''\---/''  |   |
//                    \  .-\__  `-`  ___/-. /
//                  ___`. .'  /--.--\  `. . __
//               ."" '<  `.___\_<|>_/___.'  >'"".
//              | | :  `- \`.;`\ _ /`;.`/ - ` : | |
//              \  \ `-.   \_ __\ /__ _/   .-` /  /
//         ======`-.____`-.___\_____/___.-`____.-'======
//                            `=---='
//        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
//

大佬的：

#include<iostream>
#include<cstring>
#define maxn 100005
using namespace std;
long long color[maxn][3][2],number[maxn],cnt[maxn][2],n,m,sum;
/*
之前出现过的与之同奇偶p同颜色ci的格子数设为cnt[ci][p]，
格子序号和color[ci][0][p]，格子数字和color[ci][1][p]，格子序号与数字之积之和为color[ci][2][p]
*/
int main(){
    int cor,i,j;
    scanf("%lld%lld",&n,&m);
    for(i=1;i<=n;i++) scanf("%lld",&number[i]);
    for(i=1;i<=n;i++){
        scanf("%d",&cor);
        cnt[cor][i&1]++;
        (color[cor][0][i&1]+=i)%=10007;
        (color[cor][1][i&1]+=number[i])%=10007;
        (color[cor][2][i&1]+=number[i]*i)%10007;
    }
    for(i=1;i<=m;i++)
        for(j=0;j<=1;j++){
            sum+=(color[i][0][j]*color[i][1][j])%10007+((cnt[i][j]-2)*color[i][2][j])%10007;
            sum%=10007;
        }
    printf("%lld\n",sum);
    system("pause");
    return 0;
}