大佬6的90代码求优化!(感谢调的dalao,悬关(仅限第1个调AC的))
查看原帖
大佬6的90代码求优化!(感谢调的dalao,悬关(仅限第1个调AC的))
567299
PegasusQI楼主2023/10/14 16:16
#include <bits/stdc++.h>
using namespace std;
int m,n,x[40010],cnt[40010],ans[40010][5];
int main(){
	freopen("P2119_11.in", "r", stdin);	
	freopen("P2119_11.out", "w", stdout);
	cin>>m>>n;
	for(int i=1;i<=n;i++){
		cin>>x[i];
		cnt[x[i]]++;
	}
	for(int a=1;a<=m;a++){
		for(int b=a+2;b<=m;b+=2){
			if(cnt[b]>0){
				for(int c=b+(b-a)*3+1;c<=m;c++){
					int tmp=c+(b-a)/2;
					if(cnt[c]>0&&tmp<=m&&cnt[tmp]){
						ans[a][1]+=cnt[b]*cnt[c]*cnt[tmp];
						ans[b][2]+=cnt[a]*cnt[c]*cnt[tmp];
						ans[c][3]+=cnt[a]*cnt[b]*cnt[tmp];
						ans[tmp][4]+=cnt[b]*cnt[c]*cnt[a];
					}
				}
			}
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=4;j++)cout<<ans[x[i]][j]<<" ";
		cout<<endl;
	}
	return 0;
}

成功tle。

2023/10/14 16:16
加载中...