JZOJ6412 【NOIP2019模拟11.07】极好的问题 （awesome）

极好的问题 （awesome）

题目描述：

输入：
从文件 awesome.in 中读入数据。
第一行 2 个用空格隔开的整数 n, P。
第二行 n 个用空格隔开的整数 A1, . . . , An。

输出：
输出到文件 awesome.out 中。
输出一行一个整数，表示极好的三元组的数目。

这道题是考思维和$STL$的应用。

钦定三元组$(x,y,z)$中$x<=y<=z$
首先分两种情况讨论：

1. 三元组中有重复元素
2. 三元组中没有重复元素

对于第一种情况，又分为两种情况：

1. 两个相同元素
2. 三个相同元素

然后显然三个直接判，两个的话$O(N)$枚举第三个数即可。

对于没有相同的元素，枚举$x$和$y$。
可知：$P\ast inv[P]$在$mod$意义下是$1$。
所以我们直接在$map$上找$x\ast y$的逆元。
然后判断一下这个逆元和$y$的逆元是否相同。相同减个一。

然后这样会统计两次。除个$2$即可。

/*
	事已至此，只能膜拜2017张晋杰大佬。
	stO 张晋杰 Orz 
*/
#include <map>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

typedef long long ll;

const int N = 3010;

inline void read(int &x)
{
	char ch = getchar(); x = 0;
	for(;ch < '0' || ch > '9';) ch = getchar();
	for(;ch >= '0' && ch <= '9';) x = x * 10 + (ch ^ '0'), ch = getchar();
}

#define Map map<int,int> 
Map vis; Map :: iterator it;
int n,P,a[N],b[N],m,ans,tot,inv[N];

bool cmp(int a,int b) { return a < b; }

int f_pow(int a,int x)
{
	int ret = 1;
	for(;x;x >>= 1) (x & 1) && (ret = 1ll * ret * a % P), a = 1ll * a * a % P;
	return ret % P; 
}

int main()
{
	freopen("awesome.in","r",stdin);
	freopen("awesome.out","w",stdout);
	
	read(n), read(P);
	for(int i = 1;i <= n; ++ i) read(a[i]);
	sort(a + 1,a + 1 + n,cmp);
	
	for(int i = 1;i <= n; ++ i) b[i] = a[i]; a[n + 1] = -1;
	m = unique(b + 1,b + 1 + n) - b - 1;
	for(int i = 1;i <= m; ++ i)
		inv[i] = f_pow(b[i] % P,P - 2), ++vis[inv[i]];
	
	int cur = -1,cnt = 0;
	for(int i = 1;i <= n + 1; ++ i)
		if(cur != a[i]) 
		{
			if(cnt >= 2)
				for(int j = 1;j <= m; ++ j)
					if(b[j] != cur && (1ll * cur * cur % P * b[j] % P == 1)) ++ans;
			if(cnt >= 3 && (1ll * cur * cur % P * cur % P == 1)) ++ans; 
			cur = a[i], cnt = 1;
		}
		else ++cnt;
	for(int i = 1;i <= m; ++ i)
		if(b[i] % P)
		{
			--vis[inv[i]];
			for(int j = i + 1;j <= m; ++ j)
				if(b[j] % P)
				{
					int x = 1ll * b[i] * b[j] % P;
					it = vis.find(x);
					if(it != vis.end()) tot += it -> second
					if(x == inv[j]) --tot;
				}
		}
	
	tot = tot >> 1, printf("%d",ans + tot);
	fclose(stdin); fclose(stdout); 
	return 0;
}

考场逆元那部分处理错了。直接就写了个$map$暴力30pts。
P.S：这道题写对了也就二十分（所以现在在重新评测，$GMOJ$炸了。

事已至此，只能膜拜2017张晋杰。
stO 张晋杰 Orz