提出 #73530703

---

ソースコード 拡げる

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const ll mod = 998244353;
ll n;
ll a[500050], dp1[500050], dp2[500050];
vector<ll> v[500050];

int main()
{
    cin>>n;
    for (ll i=1;i<=n;i++)
    {
        cin>>a[i];
        v[a[i]].push_back(i);
    }
    dp1[n+1]=1; dp2[n+1]=1;
    for (ll i=n;i>=1;i--)
    {
        ll add=0;
        auto idx = lower_bound(v[a[i]].begin(),v[a[i]].end(),i)-v[a[i]].begin();
        if (idx+a[i]-1<=v[a[i]].size()-1) add+=dp2[v[a[i]][idx+a[i]-1]+1];
        if (idx+a[i]<=v[a[i]].size()-1) add-=dp2[v[a[i]][idx+a[i]]+1];
        dp1[i]=(dp1[i+1]+add)%mod;
        if (i==1) break;
        ll sub=0;
        auto idx2 = lower_bound(v[a[i-1]].begin(),v[a[i-1]].end(),i)-v[a[i-1]].begin();
        if (idx2+a[i-1]-1<=v[a[i-1]].size()-1) sub=dp2[v[a[i-1]][idx2+a[i-1]-1]+1];
        dp2[i]=(dp1[i]-sub)%mod;
    }
    /*for (ll i=1;i<=n;i++) cout<<dp1[i]<<" ";
    cout<<"\n";
    for (ll i=1;i<=n;i++) cout<<dp2[i]<<" ";
    cout<<"\n";*/
    cout<<(dp1[1]-1+mod)%mod;
}

提出情報

コンパイルエラー

./Main.cpp: In function 'int main()':
./Main.cpp:23:23: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   23 |         if (idx+a[i]-1<=v[a[i]].size()-1) add+=dp2[v[a[i]][idx+a[i]-1]+1];
      |             ~~~~~~~~~~^~~~~~~~~~~~~~~~~~
./Main.cpp:24:21: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   24 |         if (idx+a[i]<=v[a[i]].size()-1) add-=dp2[v[a[i]][idx+a[i]]+1];
      |             ~~~~~~~~^~~~~~~~~~~~~~~~~~
./Main.cpp:29:26: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   29 |         if (idx2+a[i-1]-1<=v[a[i-1]].size()-1) sub=dp2[v[a[i-1]][idx2+a[i-1]-1]+1];
      |             ~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~

ジャッジ結果