結果

問題 No.1307 Rotate and Accumulate
ユーザー pppppp
提出日時 2021-01-08 15:25:03
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 700 ms / 5,000 ms
コード長 5,114 bytes
コンパイル時間 1,221 ms
コンパイル使用メモリ 109,760 KB
実行使用メモリ 46,604 KB
最終ジャッジ日時 2024-11-15 20:26:53
合計ジャッジ時間 8,638 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,820 KB
testcase_03 AC 2 ms
6,816 KB
testcase_04 AC 3 ms
6,820 KB
testcase_05 AC 2 ms
6,816 KB
testcase_06 AC 2 ms
6,816 KB
testcase_07 AC 2 ms
6,820 KB
testcase_08 AC 350 ms
27,504 KB
testcase_09 AC 352 ms
27,384 KB
testcase_10 AC 341 ms
24,708 KB
testcase_11 AC 342 ms
26,356 KB
testcase_12 AC 338 ms
24,872 KB
testcase_13 AC 47 ms
6,820 KB
testcase_14 AC 164 ms
14,120 KB
testcase_15 AC 699 ms
46,480 KB
testcase_16 AC 693 ms
46,344 KB
testcase_17 AC 700 ms
46,604 KB
testcase_18 AC 684 ms
45,572 KB
testcase_19 AC 674 ms
46,348 KB
testcase_20 AC 675 ms
45,696 KB
testcase_21 AC 1 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <fstream>
#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <string>
#include <sstream>
#include <map>
#include <set>
#include <vector>
#include <stack>
#include <cmath>
#include <queue>
#include <random>
#include <numeric>
#include <complex>

using namespace std;

#define I_MAX 2147483647
#define LL_MAX 9223372036854775807
#define ll long long
#define ld long double

struct XX{
    int l;
    int r;
    int m;
};

class xxGreater {
public:
    bool operator()(const XX& riLeft, const XX& riRight) const {
        //第2条件
        if((riLeft.r) == (riRight.r)){
            return riLeft.l < riRight.l;//<:昇順(小さいものから順番)、>:降順(大きいものから順番)
            //プライオリティキューの場合は > で、top()すると値の小さいものがとれる
        }
        //第1条件
        return (riLeft.r) < (riRight.r);
    }
};


//map<long long,long long> prime_f(long long n){
//    map<long long,long long>res;
//    for(int i=2;i*i<=n;i++){
//        while(n%i==0){
//            ++res[i];
//            n/=i;
//        }
//    }
//    if(n!=1)res[n]=1;
//    return res;
//}

//int n;
////int dat[2*10000000];
////int dat2[2*10000000];
//int dat[10];
//int dat2[10];
//
//void init(int n_){
//    n=1;
//    while(n<n_)n*=2;
//    for(int i=0;i<2*n-1;i++){
//        dat[i]=0;
//        dat2[i]=0;
//    }
//}
//
//void initset(int k,int a){
//    k+=n-1;
//    dat[k]=a;
//    while(k>0){
//        k=(k-1)/2;
//        dat[k]=dat[k*2+1]+dat[k*2+2];
//    }
//}
//
////[a,b)の間を[l,r]区間で比較しアップデート
////引数のindexに注意
////nは固定。initで計算すみ
////update2(L[i],R[i]+1,0,0,n,D[i]);
//void update2(int a,int b,int k,int l,int r,int v){//v更新値、区間は0-index
//    if(r<=a || b<=l)return;
//    if(a<=l && r<=b){
//        dat[k]+=dat2[k];
//        if(r-l>1){
//            dat2[k*2+1]+=dat2[k]/2;
//            dat2[k*2+1]+=dat2[k]/2;
//        }
//        dat2[k]=v*(r-l);
//        return;
//    }else{
//        update2(a,b,k*2+1,l,(l+r)/2,v);
//        update2(a,b,k*2+2,(l+r)/2,r,v);
//        return;
//    }
//}
//
//int query(int a,int b,int k,int l,int r){
//    if(r<=a || b<=l)return 0;
//    if(a<=l && r<=b){
//        dat[k]+=dat2[k];
//        if(r-l>1){
//            dat2[k*2+1]+=dat2[k]/2;
//            dat2[k*2+1]+=dat2[k]/2;
//        }
//        dat2[k]=0;
//        return dat[k];
//    }
//    else{
//        int vl=query(a,b,k*2+1,l,(l+r)/2);
//        int vr=query(a,b,k*2+2,(l+r)/2,r);
//        return vl+vr;
//    }
//}



//void printb(unsigned int v) {
//    unsigned int mask = (int)1 << (sizeof(v) * CHAR_BIT - 1);
//    do putchar(mask & v ? '1' : '0');
//    while (mask >>= 1);
//}

#ifdef DEBUG

#else

#endif

void dft(vector<complex<double>>& p, int inverse) {
    int sz = (int)p.size();
    if (sz == 1)return;
    vector<complex<double>> x0, x1;
    for(int i=0;i<sz/2;i++){
        x0.push_back(p[2*i]);
        x1.push_back(p[2*i+1]);
    }
    dft(x0, inverse);
    dft(x1, inverse);
    complex<double> now = 1;
    complex<double> zeta = polar(1.0, inverse * (2.0*acos(-1))/sz);//複素数の極形式　角度(2*pi)/n
    for(int i=0;i<sz;i++){
        p[i] = x0[i%(sz/2)] + now * x1[i%(sz/2)];
        now *= zeta;
    }
}
vector<double> multiply(vector<complex<double>> f, vector<complex<double>> g) {
    vector<complex<double>> p, q;
    int jisu = 1;
    while (jisu < f.size()+g.size()){
        jisu *= 2;
    }
    p.resize(jisu); q.resize(jisu);
    for(int i=0;i<f.size();i++){
        p[i] = f[i];
    }
    for(int i=0;i<g.size();i++){
        q[i] = g[i];
    }
    dft(p, 1);
    dft(q, 1);
    for(int i=0;i<jisu;i++){
        p[i] *= q[i];
    }
    dft(p, -1);
    vector<double> res;
    for(int i=0;i<jisu;i++){
        res.push_back(p[i].real()/jisu);
    }
    return res;
}

double keisu[400000];

int main(int argc, const char * argv[])
{
    //scanf("%s",S);
    //scanf("%d",&N);
    //scanf("%lld %lld",&target1,&target2);
    //sscanf(tmp.c_str(),"%dd%d%d",&time[i], &dice[i], &z[i]);
    //getline(cin, target);
    //ifstream ifs("01");//テスト用
    //ifs >> a;
    //ここから
    
    //入力高速化
    ios::sync_with_stdio(false);
    cin.tie(0);
    
    int n,q;
    cin>>n>>q;
    vector<complex<double>>f,g;
    for(int i=0;i<n;i++){
        double t1;
        cin>>t1;
        f.push_back(complex<double>(t1,0.0));
    }
    for(int i=0;i<n;i++){
        f.push_back(f[i]);
    }

    for(int i=0;i<q;i++){
        int t1;
        cin>>t1;
        keisu[n-t1]+=1.0;
    }
    for(int i=0;i<=n;i++){
        g.push_back(complex<double>(keisu[i],0.0));
    }
    vector<double> ret=multiply(f,g);
    for(int i=n;i<2*n;i++){
        printf("%.0f",ret[i]);//誤差が出るのでこの方法で出力
        printf(" ");
    }
    printf("\n");

    //ここまで
    //cout << "ans" << endl;
    //cout << " " << "ans" << endl;
    //printf("%.0f\n",ans);//小数点以下表示なし
    //printf("%.7f\n",p);
    
    return 0;
}
0