結果

問題 No.2956 Substitute with Average
ユーザー nana
提出日時 2025-01-04 17:29:55
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,541 bytes
コンパイル時間 7,872 ms
コンパイル使用メモリ 552,272 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2025-01-04 17:30:09
合計ジャッジ時間 10,987 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 6 WA * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#ifndef ONLINE_JUDGE
#define _GLIBCXX_DEBUG
#endif
#include "atcoder/all"
#include <bits/stdc++.h>
#include <climits>
#include <boost/multiprecision/cpp_dec_float.hpp>
#include <boost/multiprecision/cpp_int.hpp>
#include <boost/rational.hpp>
#include <boost/functional/hash.hpp>
// 任意長整数型
using Bint = boost::multiprecision::cpp_int;
// 仮数部が10進数で1024桁の浮動小数点数型(TLEしたら小さくする)
using BReal =
    boost::multiprecision::number<boost::multiprecision::cpp_dec_float<1024>>;
using Rat = boost::rational<Bint>;
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
// 型エイリアスの設定
template<typename T>
using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
using namespace atcoder;
template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }
template<typename T> inline bool chmin(T &a, T b) { return ((a>b) ? (a = b, true) : (false));}
#define rep(i,s,n) for(int i=s;i<(int)(n);i++)
//decrease rep(s以上, n未満)
#define drep(i,s,n) for(int i=n-1;i>=s;i--)
const long long inf = 1LL<<60;
typedef long long ll;
typedef long double ld;
typedef __int128_t l3;
//pairのsecondでソートsort(p.begin(),p.end(),cmp)
#define cmp [](pair<ll,ll> a, pair<ll,ll> b){return a.second<b.second;}
//cmpArg := 偏角ソート, atan2l(y, x); atan2l(y, x) = [-pi, pi]
#define cmpArg [](const auto &p1, const auto &p2) { return atan2l(p1.second, p1.first) < atan2l(p2.second, p2.first); }
typedef pair<int,int> pi;
typedef pair<long long, long long> P;
typedef pair<ll, pair<ll,ll> > PP;
#define vll vector<ll>
#define vvll vector<vector<ll>>
#define vint vector<int>
#define vvint vector<vector<int>>
#define vmint vector<mint>
#define vvmint vector<vector<mint>>
#define vvch vector<vector<char>>
#define vch vector<char>
#define vvpi vector<vector<pi>>
#define vpi vector<pi>
#define pb push_back
#define eb emplace_back
#define YES cout<<"Yes"<<endl;
#define NO cout<<"No"<<endl;
#define YN { cout << "Yes" << endl; } else { cout << "No" << endl; }
#define all(x) x.begin(), x.end()
#define fi first
#define se second
template<class T>istream& operator>>(istream& i, vector<T> &v) { rep(j___, 0, size(v))i >> v[j___]; return i; }
#define UNIQUE(x) sort(all(x)), x.erase(unique(all(x)), x.end())
int pc(ll x) { return __builtin_popcount(x); } //ビット列にどれだけ1がたっているかを求める pop count
void cincout(){  ios::sync_with_stdio(false);   std::cin.tie(nullptr);  cout<< fixed << setprecision(15);  }
bool out_grid(ll i, ll j, ll h, ll w) { return (!(0<=i && i<h && 0<=j && j<w)); }
#define vc_cout(v){ ll n=v.size(); rep(i___,0,n) cout << v[i___] << " "; cout << endl; }
#define vv_cout(v){ ll n=v.size(); rep(i___,0,n) { rep(j___,0,v[i___].size()){ cout << v[i___][j___] << " ";} cout << endl; }}

using mint = modint998244353;

int main()
{
    int n; cin >> n;
    vint a(n); cin >> a;
    //区間内がすべて同じもの以外はすべて異なる?
    ll ng = 0;
    rep(ave,1,31) {
        vint b(n); rep(i,0,n) b[i] = a[i] - ave;
        vint sum(n+1, 0); rep(i,1,n+1) sum[i] = sum[i-1] + b[i-1];
        int zero = 0;
        rep(i,1,n) if(sum[i] == sum[i+1] && sum[i] == 0) zero++;
        rep(l,0,n) if(sum[l] == 0) {
            if(l!=0 && sum[l+1] == 0) zero--;
            ng += zero;
        }
    }
    ll ans = ll(n)*(n+1)/2;
    ans -= ng;
    ans -= n;
    ans++;
    cout << ans << endl;
}
0