P2119 魔法阵

 题目描述  https://www.luogu.org/problemnew/show/P2119

六十年一次的魔法战争就要开始了，大魔法师准备从附近的魔法场中汲取魔法能量。

大魔法师有m个魔法物品，编号分别为,2,...,m。每个物品具有一个魔法值，我们用Xi​表示编号为i的物品的魔法值。每个魔法值Xi是不超过n的正整数，可能有多个物品的魔法值相同。

大魔法师认为，当且仅当四个编号为a,b,c,d的魔法物品满足xa​<xb​<xc​<xd​,Xb​−Xa​=2(Xd​−Xc​)，并且xb​−xa​<(xc​−xb​)/3时，这四个魔法物品形成了一个魔法阵，他称这四个魔法物品分别为这个魔法阵的A物品，B物品，C物品，D物品。

现在，大魔法师想要知道，对于每个魔法物品，作为某个魔法阵的A物品出现的次数，作为B物品的次数，作为C物品的次数，和作为D物品的次数。

输入输出格式

输入格式：

 

第一行包含两个空格隔开的正整数n,m。

接下来m行，每行一个正整数，第i+1行的正整数表示Xi​，即编号为i的物品的魔法值

 

输出格式：

 

共m行，每行4个整数。第ii行的4个整数依次表示编号为ii的物品作 为A,B,C,D物品分别出现的次数。

输入输出样例

输入样例#1： 复制

30 8
1
24
7
28
5
29
26
24

输出样例#1： 复制

4 0 0 0
0 0 1 0
0 2 0 0
0 0 1 1
1 3 0 0
0 0 0 2
0 0 2 2
0 0 1 0

采用中间相遇方法。也是过程的真实模拟！虽然得了80分，但是我觉得方法很不错。

#include <iostream>
#include<queue>
#include<set>
#include<vector>
#include<string.h>
#include<cstdio>
#include<ctime>
using namespace std;
int n,m,M;
vector< vector<int> > lv,rv;
vector<int> times;
inline int read(){
    int x=0,f=1;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')f=-f;c=getchar();}
    while(c>='0'&&c<='9'){x=(x<<1)+(x<<3)+c-'0';c=getchar();}
    return x*f;
}
int main() {
	cin>>n>>m;
	M=m;
	int n2=(n+2)/2;
	lv.resize(n2);
	rv.resize(n2);
	vector<int> times(n+1,0);
	vector<int> vaip(m);
	set<int> vat;

	vector<int> ttp(4,0);
	vector< vector<int> > res(n+1,ttp);
    clock_t start=clock(),end;
    double endtime;
	for(int i=0; i<m; i++) {
		vaip[i]=read();
		if(!vat.count(vaip[i]))
			vat.insert(vaip[i]);
		times[vaip[i]]++;
	}
	m=vat.size();
	vector<int> va(vat.begin(),vat.end());
    
	for(int i=0; i<m; i++) {
		int bb=va[i],cha;
		for(int j=0; j<i; j++) {
			cha=bb-va[j];
			if(cha%2==0) {
				lv[cha/2].push_back(bb);
			}
		}
	}

	for(int i=0; i<m; i++) {
		int bb=va[i],cha;
		for(int j=0; j<i; j++) {
			cha=bb-va[j];
			if(cha<n2) {
				rv[cha].push_back(bb);
			}
		}
	}
    
	int pa,pb,pc,pd;
	for(int i=0; i<n2; i++) {
		if(rv[i].size()&&lv[i].size()) {
			for(int j=0; j<lv[i].size(); j++) {
				pb=lv[i][j];
				pa=pb-2*i;
				for(int k=0; k<rv[i].size(); k++) {
					pd=rv[i][k];
					pc=pd-i;
					if(pb<pc&&4*pb<pc+pa*3) {
						res[pa][0]+=times[pb]*times[pc]*times[pd];
						res[pb][1]+=times[pa]*times[pc]*times[pd];
						res[pc][2]+=times[pa]*times[pb]*times[pd];
						res[pd][3]+=times[pa]*times[pb]*times[pc];
					}
				}
			}

		}
	}
  
//	for(int i=0; i<vaip.size(); i++) {
//		printf("%d %d %d %d \n",res[vaip[i]][0],res[vaip[i]][1],res[vaip[i]][2],res[vaip[i]][3]);
//	}
   for(int i=0;i<M;i++)
    {
       for(int j=0;j<4;j++)
       cout<<res[vaip[i]][j]<<" ";
       cout<<endl;	
    }
	endtime=(double)(clock()-start)/CLOCKS_PER_SEC;
    cout<<"Total time:"<<endtime<<endl;
	return 0;
}