P2671 [NOIP2015 普及組] 求和

原題連結

題解

得分有哪些條件？顏色相同，下標同奇或同偶，
由此我們對數進行sort分類，把顏色相同的放一起，然後顏色內部按下標小到大排
然後就是很關鍵的分數計算公式
\((x + z) \times (\text{number}_x + \text{number}_z).\)
可以分解為
\(\text{number}_x x + \text{number}_x z + \text{number}_z x + \text{number}_z z\)
這下就很明瞭了，$\text{number}_x x $出現的次數為與x同性的數字的次數，而所有的 \(\text{number}_z x\) 可以和為 \(x · Sum-number_x\)

只要細心，肯定可以想出來的！！

code

#include<bits/stdc++.h>
using namespace std;
const int mod=10007;
struct node
{
    int score,color,x;
}num[100007];

bool cmp(node a,node b)
{
    if(a.color!=b.color) return a.color<b.color;
    return a.x<b.x;
}
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>num[i].score;

    for(int i=1;i<=n;i++)
    {
        cin>>num[i].color;
        num[i].x=i;
    }

    sort(num+1,num+1+n,cmp);
    int j,i=1;
    int ans=0;
    for(;i<=n;)
    {
        int odd=0,even=0,oddsum=0,evensum=0;
        for(j=i;num[j].color==num[i].color;j++)
        {
            if(num[j].x%2==0) even++,evensum+=num[j].score;
            else odd++,oddsum+=num[j].score;
            evensum%=mod;
            oddsum%=mod;
        }
        for(int l=i;l<j;l++)
        {
            if(num[l].x%2==0)
            {
                ans=ans+(num[l].x*((even-1)*num[l].score+evensum-num[l].score));
                ans%=mod;
            }
            else
            {
                ans=ans+(num[l].x*((odd-1)*num[l].score+oddsum-num[l].score));
                ans%=mod;
            }
        }
        i=j;
    }

    cout<<ans;
    return 0;
}