提出 #27234105

---

ソースコード 拡げる

#include <bits/stdc++.h>
using namespace std;
#define M 1000000007
#define U 998244353
#define N 1000005
#define int long long
#define sz(c) (int)c.size()
#define fr first
#define ll long long 
#define sc second
#define pb push_back
#define ppb pop_back
#define mp make_pair
#define all(a) (a).begin(),(a).end()
#define rep(i,a,n) for(int i=a ; i<n ; i++)
#define r0 return 0;
#define endl '\n'
#define INF (int)1e15

/* Debug Begins */ 
# define trace(args...) { string _s = #args; replace(_s.begin(), _s.end(), ',', ' '); \
  stringstream _ss(_s); istream_iterator<string> _it(_ss); err(_it, args); }
string to_string(char c) { return "'" + string(1, c) + "'";}
string to_string(string s) { return '"' + s + '"';}
string to_string(bool f) { if(f) return "True"; else return "False";}
string to_string(const char* s) { return to_string((string) s);}
template<typename A> string to_string(A);
template<typename A, typename B> string to_string(pair<A, B> p){
  return "(" + to_string(p.first) + ": " + to_string(p.second) + ")";}
template<typename A> string to_string(A v) {bool f = false; string r = "{"; 
  for (auto x: v) {if (f) r += ", "; r += to_string(x); f = true;} return r += "}";}
template<typename A> string to_string(vector<vector<A>> v) {string r; 
  for (auto x: v) r += "\n" + to_string(x); return r;}
int Nerr;
template<typename A> string to_string(A *p) {return to_string(vector<A>(p, p + Nerr));}
void err(istream_iterator<string>) { cerr << endl; }
template<typename T,typename... Args> void err(istream_iterator<string> it, T a, Args... args) {
  cerr << *it << " = " << to_string(a) << "; "; err(++it, args...); }
template<typename T> void kek(T ans) {cout << ans << endl; exit(0);}
#define Lu(...) [&] (auto &&u) { return __VA_ARGS__; }
#define Luv(...) [&] (auto &&u, auto &&v) { return __VA_ARGS__; }
/***************************************************************/

vector<int> mark(N, false);
vector<long long> primes;

vector<vector<int>> segmentedSieve(long long L, long long R) {
    long long lim = sqrt(R);
    vector<vector<int>> isPrime(R - L + 1);
    for (long long i : primes){
        for (long long j = max(i * i, (L + i - 1) / i * i); j <= R; j += i){
            isPrime[j - L].push_back(i);
        }
    }
    if (L == 1)
        isPrime[0] = {1};
    return isPrime;
}


signed main()
{
    ios_base::sync_with_stdio(0);
    int TESTS=1;
    for(int i = 2; i < N; i++)
        mark[i] = i;
    for (long long i = 2; i < N; ++i) {
        if (mark[i] == i) {
            primes.emplace_back(i);
            for (long long j = i * i; j < N; j += i)
                mark[j] = i;
        }
    }
    // cin>>TESTS;
    while(TESTS--)
    {   
        int n, k;
        cin >> n >> k;
        if(k == 0 or k == n) {
            cout << 1 << endl;
            continue;
        }
        unordered_map<int,int> l;
        for(int i = 1; i <= k; i++) {
            int t = i;
            while(t > 1) {
                // trace(t, mark[t]);
                l[mark[t]]--;
                t /= mark[t];
            }
        }
        // trace(l);
        auto p = segmentedSieve(n-k+1, n);
        for(int i = 0; i < k; i++) {
            vector<int> pri = p[i];
            int temp = n-k+1+i;
            for(auto i : pri) {
                while(temp % i == 0) {
                    temp /= i;
                    l[i]++;
                }
            }
            if(temp > 1) l[temp]++;
        }
        // trace(l);
        int ans = 1;
        for(auto j : l) {
            ans = (ans * (j.sc + 1)) % U;
        }
        cout << ans << endl;
    }
}

提出情報

コンパイルエラー

./Main.cpp: In function ‘std::string to_string(std::vector<std::vector<_Tp> >)’:
./Main.cpp:33:3: warning: this ‘for’ clause does not guard... [-Wmisleading-indentation]
   33 |   for (auto x: v) r += "\n" + to_string(x); return r;}
      |   ^~~
./Main.cpp:33:45: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the ‘for’
   33 |   for (auto x: v) r += "\n" + to_string(x); return r;}
      |                                             ^~~~~~
./Main.cpp: In function ‘std::vector<std::vector<long long int> > segmentedSieve(long long int, long long int)’:
./Main.cpp:48:15: warning: unused variable ‘lim’ [-Wunused-variable]
   48 |     long long lim = sqrt(R);
      |               ^~~

ジャッジ結果