題目描述 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;
}