#857450 (C++17(clang)) No.2260 Adic Sum

提出ソース

結果

問題	No.2260 Adic Sum
ユーザー	georgejsh
提出日時	2023-04-07 22:18:01
言語	C++17(clang) (17.0.6 + boost 1.83.0)
結果	AC
実行時間	924 ms / 2,000 ms
コード長	5,708 bytes
コンパイル時間	2,530 ms
コンパイル使用メモリ	165,760 KB
実行使用メモリ	113,280 KB
最終ジャッジ日時	2024-10-06 07:18:37
合計ジャッジ時間	8,738 ms
ジャッジサーバーID （参考情報）	judge4 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	1 ms 5,248 KB
testcase_02	AC	1 ms 5,248 KB
testcase_03	AC	1 ms 5,248 KB
testcase_04	AC	1 ms 5,248 KB
testcase_05	AC	1 ms 5,248 KB
testcase_06	AC	2 ms 5,248 KB
testcase_07	AC	1 ms 5,248 KB
testcase_08	AC	2 ms 5,248 KB
testcase_09	AC	2 ms 5,248 KB
testcase_10	AC	2 ms 5,248 KB
testcase_11	AC	2 ms 5,248 KB
testcase_12	AC	3 ms 5,248 KB
testcase_13	AC	3 ms 5,248 KB
testcase_14	AC	18 ms 5,888 KB
testcase_15	AC	25 ms 7,040 KB
testcase_16	AC	14 ms 5,524 KB
testcase_17	AC	27 ms 7,296 KB
testcase_18	AC	513 ms 64,128 KB
testcase_19	AC	558 ms 68,224 KB
testcase_20	AC	435 ms 49,536 KB
testcase_21	AC	436 ms 50,176 KB
testcase_22	AC	325 ms 38,400 KB
testcase_23	AC	150 ms 20,864 KB
testcase_24	AC	545 ms 63,744 KB
testcase_25	AC	46 ms 9,600 KB
testcase_26	AC	46 ms 9,472 KB
testcase_27	AC	48 ms 9,856 KB
testcase_28	AC	45 ms 9,600 KB
testcase_29	AC	44 ms 9,728 KB
testcase_30	AC	42 ms 7,936 KB
testcase_31	AC	217 ms 37,248 KB
testcase_32	AC	225 ms 37,120 KB
testcase_33	AC	48 ms 10,624 KB
testcase_34	AC	183 ms 30,848 KB
testcase_35	AC	924 ms 113,280 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>


using namespace std;


typedef long long int ll;
typedef long double ld;

typedef pair<ll, ll> pi;
typedef pair<ld,ld> pd;

typedef vector<ll> vi;
typedef vector<ld> vd;
typedef vector<pi> vpi;

#define repab(i, a, b) for (ll i = (a); i < (b); i++)
#define repabs(i, a, b,s) for (ll i = (a); i < (b); i+=(s))
#define rep(i, a) for (ll i = 0; i < (a); i++)
#define rrepab(i,a,b) for (ll i = (b)-1; i >= (a); i--)
#define rrepabs(i,a,b,s) for (ll i = (b)-1; i >= (a); i-=(s))
#define rrep(i,a) for (ll i = (a)-1; i >= 0; i--)
#define iter(a, x) for (auto& a : x)
#define repeat(i, begin, end) for (__typeof(end) i = (begin) - ((begin) > (end)); i != (end) - ((begin) > (end)); i += 1 - 2 * ((begin) > (end)))
#define GET5(a, b, c, d, e, ...) e
#define F_ORC(...) GET5(__VA_ARGS__, repabs, repab, rep)
#define F_ORRC(...) GET5(__VA_ARGS__, rrepabs, rrepab, rrep)
#define FOR(...) F_ORC(__VA_ARGS__)(__VA_ARGS__)
#define FORR(...) F_ORRC(__VA_ARGS__)(__VA_ARGS__)
#define v(a) vector<array<ll,a>>
#define minheap(a) priority_queue<array<ll,a>,vector<array<ll,a>>,greater<array<ll,a>> >

#define rloop(n) int n;re(n);rep(i,n)
#define array(n) array<ll,n>
#define rea(arr,n) int n;re(n);vi arr(n);re(arr);

#define mk make_pair
#define pb push_back
#define fi first
#define se second
#define lb lower_bound
#define ub upper_bound

#define sz(x) (int)x.size()
#define beg(x) x.begin()
#define en(x) x.end()
#define all(x) beg(x), en(x)
#define resz resize


template<class T> void ckmin(T &a, T b) { a = min(a, b); }
template<class T> void ckmax(T &a, T b) { a = max(a, b); }

namespace io {
     #define error(args...) { string _s = #args; replace(_s.begin(), _s.end(), ',', ' '); stringstream _ss(_s); istream_iterator<string> _it(_ss);cerr<< "line: "<<__LINE__ <<" "; err(_it, args); cerr<<endl;}

    void err(istream_iterator<string> it) {}
    
    
   
    // TYPE ID (StackOverflow)
    template<class T> struct like_array : is_array<T>{};
    template<class T, size_t N> struct like_array<array<T,N>> : true_type{};
    template<class T> struct like_array<vector<T>> : true_type{};
    template<class T> bool is_like_array(const T& a) { return like_array<T>::value; }
    
    
    // INPUT 
    
    template<class T> void re(T& x) { cin >> x; }
    template<class Arg, class... Args> void re(Arg& first, Args&... rest);
    void re(double& x) { string t; re(t); x = stod(t); }
    void re(ld& x) { string t; re(t); x = stold(t); }
    
    template<class T1, class T2> void re(pair<T1,T2>& p);
    template<class T> void re(vector<T>& a);
    template<class T, size_t SZ> void re(array<T,SZ>& a);
    
    template<class Arg, class... Args> void re(Arg& first, Args&... rest) { re(first); re(rest...); }
    template<class T1, class T2> void re(pair<T1,T2>& p) { re(p.fi,p.se); }
    template<class T> void re(vector<T>& a) { rep(i,sz(a)) re(a[i]); }
    template<class T, size_t SZ> void re(array<T,SZ>& a) { rep(i,SZ) re(a[i]); }
    template<class T, size_t SZ> ostream& operator<<(ostream& os, const array<T,SZ>& a);
    template<class T> ostream& operator<<(ostream& os, const vector<T>& a);
    template<class T> ostream& operator<<(ostream& os, const set<T>& a) ;
    template<class T1, class T2> ostream& operator<<(ostream& os, const map<T1,T2>& a) ;
    // OUTPUT 
    
    template<class T1, class T2> ostream& operator<<(ostream& os, const pair<T1,T2>& a) {
        os << '{' << a.fi << ", " << a.se << '}'; return os;
    }
    template<class T> ostream& printArray(ostream& os, const T& a, int SZ) {
        os << '{';
        rep(i,SZ) {
            if (i) {
                os << ", ";
                if (is_like_array(a[i])) cout << " ";
            }
            os << a[i];
        }
        os << '}';
        return os;
    }
    template<class T, size_t SZ> ostream& operator<<(ostream& os, const array<T,SZ>& a) {
        return printArray(os,a,SZ);
    }
    template<class T> ostream& operator<<(ostream& os, const vector<T>& a) {
        return printArray(os,a,sz(a));
    }
    template<class T> ostream& operator<<(ostream& os, const set<T>& a) {
        os << vector<T>(all(a)); return os;
    }
    template<class T1, class T2> ostream& operator<<(ostream& os, const map<T1,T2>& a) {
        os << vector<pair<T1,T2>>(all(a)); return os;
    }
    
    template<class T> void pr(const T& x) { cout << x << '\n'; }
    template<class Arg, class... Args> void pr(const Arg& first, const Args&... rest) { 
        cout << first << ' '; pr(rest...); 
    }
    template<typename T, typename... Args>
    void err(istream_iterator<string> it, T a, Args... args) {
        cerr << *it << " = " << a<<" " ;
        err(++it, args...);
    }
    template<typename T>
    void pa(vector<T> &a,bool psize=true){
        if(psize) cout<<a.size()<<"\n";
        iter(x,a) cout<<x<<" ";
        cout<<"\n";
    }
}
using namespace io;

const int MOD = 1000000007;
//const int MOD = 998244353;
const ll INF = 1e18;
const int MX = 2e5+1;

#define yes cout<<"YES\n"
#define no cout<<"NO\n"

void solve(){
   ll n,p;re(n,p);
   vi a(n);
   re(a);
   vi pos;
   pos.pb(p);
   while(pos.back()<1e10) {
    pos.pb(pos.back()*p);
   }
   pos.pop_back();
   //error(pos);
   map<ll,ll> mp[sz(pos)];
    rep(j,sz(pos))rep(i,n) mp[j][a[i]%pos[j]]++;
    ll curr=0;
    ll ans=0;
    rrep(i,sz(pos)){
        ans-=curr*(i+1);
        curr=0;
        //error(i,mp[i]);
        iter(y,mp[i]) {
            curr+=y.se*(y.se-1)/2;
            ans+=y.se*(y.se-1)/2*(i+1);
        }
       // error(curr,ans);           
    }
    cout<<ans<<"\n";
}
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    //rloop(t){
        solve();
    //}
    return 0;
}

yukicoder

結果

テストケース

ソースコード