洛谷 P2671 求和（vector，数论

P2671 求和
##20分做法

看完这题，第一想法当然是无脑暴力啦…直接枚举x,y,z,看是否满足条件即可。算法复杂度为O(N^3)

这样就可得20分了。当然，如果你想用更高级的算法不开long long也是可以的。

##40分做法

可以直接枚举x，z的值，通过条件(1)求出y。再看是否满足条件。算法复杂度为O(N^2)

##40~50分做法

仍然是枚举x,z的值，但可以先分析x+z=2*y的奇偶性，因为xyz是整数，因此2y是2的倍数，因此x,z必然都为偶数或奇数，因此可以分奇偶性进行枚举，此时这个三元组即可不考虑y值的大小，即为当(i%2==1)枚举前面奇数序列相同的颜色进行算分数即可，算法复杂度为O(N^2/2)
因为要枚举两个数的值（x,y），所有无论如何一定会TLE。
通过简单的分析：

• xyz是整数,x<y<z,y−x=z−y
• colorx=colorz

所以x,y的奇偶性一定相同，那么通过开两个vector组，分别存奇数和偶数各个颜色的标号。由于每个颜色的vector存入的元素个数可能不大，所以计算出分数和的时间较小，有可能不超时。。
注意：sum必须要用long long。

#include <iostream>
#include <algorithm>
#include <cstring>
#include <math.h>
#include <functional>
#include <string>
#include <vector>
#include <map>
#include <stack>
#include <cstdio>
#include <list>
#include <queue>
#include <set>
#include <unordered_map>
#define ll long long
using namespace std;


vector <ll> col1[100005],col2[100005];
ll n,m;
ll sum=0;
typedef struct{
   
    ll num,color;
}node;
node k[100005];
/*
 * 1 1 1
 * 1 0 1 0 1
 * 1 0 0 1 0 0 1
 * 1 0 0 0 1 0 0 0 1
 *  */
int main(){
   
    cin>>n>>m;
    ll i;
    for(i=0;i<n;i++) cin>>k[i].num;
    for(i=0;i<n;i++) cin>>k[i].color;
    for