题解 | Integration-2019牛客暑期多校训练营第一场B题

最新推荐文章于 2020-07-14 09:28:44 发布

原创最新推荐文章于 2020-07-14 09:28:44 发布 · 215 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#题解 #Integration #19牛客暑期多校训练营

收录于

题解同时被 2 个专栏收录

97 篇文章

订阅专栏

19牛客暑期多校训练营第一场

3 篇文章

订阅专栏

本文解析了牛客竞赛中的一道算法题，通过逆元求解特定数学问题，提供了详细的代码实现，包括输入处理、逆元计算及最终结果输出。

题目来源于牛客竞赛：https://ac.nowcoder.com/acm/contest/discuss
题目描述：
在这里插入图片描述
输入描述：

输出描述：

示例1：

题解：

代码：

#include <bits/stdc++.h>

static const int MOD = 1e9 + 7;

int inverse(int a)
{
  return a == 1 ? 1 : 1LL * (MOD - MOD / a) * inverse(MOD % a) % MOD;
}

int main() {
  int n;
  while (scanf("%d", &n) == 1) {
    std::vector<int> a(n);
    for (int i = 0; i < n; ++i) {
      scanf("%d", &a[i]);
    }
    int result = 0;
    for (int i = 0; i < n; ++i) {
      int p = 2LL * a[i] % MOD;
      int aa = 1LL * a[i] * a[i] % MOD;
      for (int j = 0; j < n; ++j) {
        if (i != j) {
          p = 1LL * p * (1LL * a[j] * a[j] % MOD + MOD - aa) % MOD;
        }
      }
      (result += inverse(p)) %= MOD;
    }
    printf("%d\n", result);
  }
}

更多问题可关注牛客竞赛区，一个刷题、比赛、分享的社区。
传送门：https://ac.nowcoder.com/acm/contest/discuss