2018搜狐秋招 Kolakoski生成序列问题答案

原创于 2017-08-28 21:12:20 发布 · 300 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #算法 #搜狐

本文介绍了一个用于生成科拉科斯基序列的Java程序。该程序通过输入初始数组和序列长度来计算并输出指定长度的科拉科斯基序列。使用了动态规划的方法来解决这一数学问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package SouHu;
import java.util.Scanner;

public class Kolakoski_Sequence {
    public int[] solve(int[] arr,int n){
        int[] dp=new int[n];
        int i=0;
        int cur=0;
        while(i<n){
            for(int p=0;p<arr.length;p++){
                int c=dp[cur];
                if(c==0) c=arr[p];
                while(c-->0){
                    dp[i++]=arr[p];
                    if(i==n) return dp;
                }
                cur++;
            }
        }

        return dp;
    }

    public static void main(String[] args){
        Kolakoski_Sequence s=new Kolakoski_Sequence();
        Scanner in=new Scanner(System.in);

        int[] arr;
        while(in.hasNext()){
            int n=in.nextInt();
            int m=in.nextInt();
            arr=new int[m];
            for(int i=0;i<m;i++) arr[i]=in.nextInt();
            int[] res=s.solve(arr,n);
            for(int i=0;i<n;i++){
                System.out.println(res[i]);
            }
        }
    }
}