1. 程式人生 > >AGC003E Sequential operations on Sequence

AGC003E Sequential operations on Sequence

好題 方法 有一個 lin algo inline 表示 stk lld

題意簡述:一串數,初始為\(1~N\),現有\(Q\)個操作,每次操作會把數組長度變成\(L_i\),多余的長度直接截斷;長度不夠則循環填充,問最後\(1~N\)每個數的出現次數。

思維好題啊,跪了。

首先我們很容易發現,如果存在\(i<j,L_i>L_j\)這種情況,那麽\(L_i\)就沒有什麽用了,所以我們可以把詢問搞成一個單調遞增的序列。

然後我們有一個非常巧妙的方法,我們反著考慮,設\(f_i\)表示當前串在最後狀態出現的次數,那麽\(f_n=1\)

然後反著做,對於每個詢問i,\(f[i]+=\lfloor\frac{q[i+1]}{q[i]}\rfloor*f[i+1]\)

,余數部分遞歸處理,最後一定會得到一個小於\(min_{L_i}\)的部分,直接是由原序列轉移過來的,將答案差分處理即可。

註意一開始要把n入棧,因為不這麽搞統計答案時如果\(min_{L_i}>n\),就越界了。

#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1e5+10;
#define ll long long
ll stk[N],f[N],ans[N];
int n,top,q;
void solve(ll x,ll v){
    int k=upper_bound(stk+1,stk+top+1,x)-stk-1;
    if(!k){
        ans[1]+=v;ans[x+1]-=v;
        return ;
    }
    f[k]+=v*(x/stk[k]);solve(x%stk[k],v);
}
int main(){
    scanf("%d%d",&n,&q);stk[++top]=n;
    for(int i=1;i<=q;i++){
        ll x;scanf("%lld",&x);
        while(x<=stk[top]&&top)--top;
        stk[++top]=x;
    }
    f[top]=1;
    for(int i=top-1;i;i--)
        f[i]+=f[i+1]*(stk[i+1]/stk[i]),solve(stk[i+1]%stk[i],f[i+1]);
    ans[1]+=f[1];ans[stk[1]+1]-=f[1];
    for(int i=1;i<=n;i++)printf("%lld\n",ans[i]+=ans[i-1]);
}

AGC003E Sequential operations on Sequence