2017 ACM Jordanian Collegiate J.Efficiency Test 动态规划、类倍增

题目

题目链接

题解及代码

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 1e5+7;
int f[2*N][55],a[2*N],sum[2*N],ans[2*N],pb[2*N];
int T,n,k;
#define pr(x) cout<<#x<<":"<<x<<endl
/** 题解
 * 由于最多跳一个环就结束，因此倍长数组就足够了。
 * 预处理出f[i][j]表示从i点出发，每次跳j步，最远能跳到哪里。
 * 预处理出sum[i]表示数组[1,i]位置中有多少个'#'。
 * 预处理出pb[i]，表示i位置前面距离最近的'#'的位置。
 * ans数组为存储最终答案的数组。
 * 枚举出发点i，再从大到小枚举跳跃长度m，很容易得到f[i][m]得到下一个点nxt，
 * 如果nxt不为0，则表示可以走，为了避免多跳，我们把nxt和pb[i]+1取一个最小值，
 * 然后ans[i] += ceil((nxt-now)/m);
 * 并更新now = nxt;
 * 这样的话，总的时间复杂度就是O(nk)
 */
int main(){
    freopen("jumps.in","r",stdin);
    cin>>T;
    while(T--){
        cin>>n>>k;
        memset(f,0,sizeof(f));
        memset(sum,0,sizeof(sum));
        memset(a,0,sizeof(a));
        memset(ans,0,sizeof(ans));

        for(int i = 1;i <= n;++i){
            char c;
            scanf(" %c",&c);
            if(c == '#') a[i] = 1;
            sum[i] = sum[i-1] + a[i];
        }
        for(int i = n+1;i <= 2*n;++i){
            a[i] = a[i-n];
            sum[i] = sum[i-1] + a[i];
        }
        sum[2*n+1] = sum[2*n];

        int pre = 0;
        for(int i = 1;i <= 2*n;++i){
            if(a[i]) pre = i;
            else pb[i] = pre;
        }

        for(int i = 2*n;i;--i) for(int m = k;m;--m){
            if(a[i]) continue;
            int t = m + i;
            if(t > 2*n){
                if(!a[t-2*n] && sum[2*n]-sum[i]+sum[t-2*n] <= 1)
                    f[i][m] = 2*n+1;
            }
            else{
                if(!a[t] && sum[t]-sum[i] <= 1)
                    f[i][m] = max(f[t][m],t);
            }
        }

        for(int i = 1;i <= n;++i){
            if(a[i]) continue;
            int it = i;
            for(int m = k;m;--m){
                int nxt = f[it][m];
                if(nxt) {
                    if(nxt > pb[i+n]) nxt = pb[i+n]+1;
                    ans[i] += (nxt-it) / m;
                    if((nxt-it)%m) ans[i]++;
                    it = nxt;
                }
                if(it > pb[i+n]) break;

            }
            if(it < pb[i+n]) ans[i] = -1;
        }
        for(int i = 1;i <= n;++i){
            if(i != 1) cout<<' ';
            cout<<ans[i];
        }
        cout<<endl;
    }
    return 0;
}