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;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章