結果

問題 No.1261 数字集め
ユーザー chocorusk
提出日時 2020-10-21 00:23:56
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 2,502 bytes
コンパイル時間 4,702 ms
コンパイル使用メモリ 186,316 KB
最終ジャッジ日時 2025-01-15 12:05:19
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 3 WA * 3 RE * 88
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#include <list>
#include <atcoder/all>
#define popcount __builtin_popcount
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef pair<int, int> P;
using mint=modint1000000007;
bool e[11000][11000];
int main()
{
    int n, m;
    cin>>n>>m;
    mint a[100010];
    for(int i=2; i<=n; i++){
        int ai; cin>>ai;
        ai--;
        a[i]=mint(ai);
    }
    int q; cin>>q;
    vector<int> v;
    for(int i=2; i<n; i++){
        if(n%i==0) v.push_back(i);
    }
    mint ap[100010];
    for(int i=2; i<=n; i++) ap[i]=a[i].pow(m-1);
    auto calc=[&](int i, int j){
        return (ap[i]*a[n]-ap[i]*a[j]+ap[j]*a[i]-ap[j]*a[n]+ap[n]*a[j]-ap[n]*a[i])/((a[i]-a[n])*(a[i]-a[j])*(a[n]-a[j]));
    };
    mint ans=0;
    for(auto x:v){
        for(auto y:v){
            if(x<y && y%x==0){
                ans+=calc(x, y);
            }
        }
    }
    cout<<ans.val()<<endl;
    v.push_back(1);
    v.push_back(n);
    int x[10010], y[10010];
    for(int i=0; i<q; i++){
        cin>>x[i]>>y[i];
        v.push_back(x[i]);
        v.push_back(y[i]);
    }
    sort(v.begin(), v.end());
    v.erase(unique(v.begin(), v.end()), v.end());
    int l=v.size();
    for(int i=0; i<l; i++){
        for(int j=0; j<i; j++){
            if(v[i]%v[j]==0){
                e[j][i]=1;
            }
        }
    }
    for(int i=0; i<q; i++){
        int xn=lower_bound(v.begin(), v.end(), x[i])-v.begin();
        int yn=lower_bound(v.begin(), v.end(), y[i])-v.begin();
        if(x[i]==1 && y[i]<n){
            for(int j=yn+1; j<l; j++){
                if(e[yn][j] && e[j][l-1]){
                    ans+=calc(y[i], v[j]);
                }
            }
        }else if(x[i]<n && y[i]==n){
            for(int j=1; j<xn; j++){
                if(e[0][j] && e[j][xn]){
                    ans+=calc(v[j], x[i]);
                }
            }
        }else if(1<x[i] && y[i]<n){
            if(e[0][xn] && e[yn][l-1]){
                ans+=calc(x[i], y[i]);
            }
        }
        e[xn][yn]=1;
        cout<<ans.val()<<endl;
    }
    return 0;
}
0