首页 > 代码库 > POJ1840 hash
POJ1840 hash
POJ1840
问题重述:
给定系数a1,a2, ..,a5,求满足a1 * x1 ^ 3 + a2 * x2 ^ 3 +... + a5 * x5 ^ 3 = 0的 xi 的组数。其中ai, xi都在[-50, 50]内,且xi != 0。
算法:
1)用h[i]记录满足a1 * x1 ^3 + a2 * x2 ^ 3 = i 的x1, x2组数。
2)令ans = 0, 循环x1, x2, x3的值,若 tmp = a3 * x3^3 + a4 * x4^3 + a5 * x5^3, h[-tmp] != 0, 则 ans += h[-tmp]。
AC代码
1 #include <iostream> 2 #include <cstring> 3 #include <cstdio> 4 5 using namespace std; 6 7 const int maxn = 25000010; 8 const int offset = maxn / 2; 9 short h[maxn];10 int a[5];11 12 void _hash()13 {14 memset( h, 0, sizeof(h) );15 for (int i = -50; i <= 50; i++) {16 if (i == 0) continue;17 for (int j = -50; j <= 50; j++) {18 if (j == 0) continue;19 h[a[0] * i * i * i + a[1] * j * j * j + offset]++;20 }21 }22 }23 24 25 int main()26 {27 for (int i = 0; i < 5; i++)28 cin >> a[i];29 _hash();30 int ans = 0;31 for (int i = -50; i <= 50; i++) 32 for (int j = -50; j <= 50; j++)33 for (int k = -50; k <= 50; k++) {34 if (i * j * k != 0) {35 int tmp = a[2] * i * i * i + 36 a[3] * j * j * j +37 a[4] * k * k * k;38 if (offset - tmp < maxn && offset - tmp >=0)39 if ( h[offset - tmp] != 0)40 ans += h[offset - tmp];41 }42 }43 cout << ans << endl;44 return 0;45 }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。