第二题答案,可能不是最终答案,但整体思路没动过 #include <iostream> #include <vector> #include <unordered_map> #include <cmath> #define LL long long using namespace std; int main() { int n; cin >> n; vector<unordered_map<int,int>> t(9, unordered_map<int,int>()); vector<int> nums; for (int i = 0; i < n; ++i) { LL x; cin >> x; nums.push_back(x); LL base = 10; for (int j = 0; j < 9; ++j) { t[j][int((LL)(x)*(LL)(base) % (LL)(7))]++; base *= 10; } } int cnt = 0; for (int i = 0; i < n; ++i) { int base = 0; int x = nums[i]; do { base++; x /= 10; } while (x > 0); int remain = (7 - nums[i] % 7) % 7; if (t[base-1][remain] >= 2) if (((LL)(nums[i]) * (LL)(pow(10, base))) % 7 == remain) { cnt += t[base - 1][remain] - 1; } else cnt += t[base - 1][remain]; else if (t[base-1][remain] < 1) continue; else { if (((LL)(nums[i]) * (LL)(pow(10, base))) % 7 == remain) { continue; } else cnt++; } } cout << cnt << endl; return 0; }
点赞 1

相关推荐

牛客网
牛客企业服务