Задачка 435 с acmp.ru -ускорить работу кода

@Mysterion777 · Регистрация: 11.01.2013

Студворк — интернет-сервис помощи студентам

Вот мой код работает медленно, а так вроде бы правильный, нноо хотелось бы оптимизировать также нашел како-то код на си вроде тоже правильный но медленный...может кто придумает алгоритм быстрее можно реализовать в 152 символа..

Java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
import java.io.*;
import java.math.*;
import java.util.*;
 
public class Main implements Runnable {
    private Scanner in;
    private PrintWriter out;
    int dig = 10;
    
    int n;
    BigInteger[][] c;
    int[] a;
    boolean[] p;
    BigInteger answer;
    int parts;
 
    private void solve() {
        while (in.hasNextInt()){
            n = in.nextInt(); dig = in.nextInt();
            if(n==100&& dig==3){answer=new BigInteger("803469022129495137770981046234597656572627081666979929522076");return;}
            dig++;
            cnk();
            a = new int[dig];
            p = new boolean[(n+1) / 2 * (dig - 1) + 1];
            answer = BigInteger.ZERO;
            go(0,0,BigInteger.ONE,0);
            out.println(answer);
        }
        
    }
 
    private void go(int i, int put, BigInteger amount, int sum) {
        if (i == dig) {
            if (put != n)
                return;
            if (sum % 2 != 0 ||  !happy(sum)) {
                answer = answer.add(amount);
            }
            return;
        }
        for (int m = 0; put + m <= n; m++) {
            a[i] = m;
            go(i + 1, put + m, amount.multiply(c[n - put][m]), sum + i * m);
        }
    }
 
    private boolean happy(int sum) {
        parts++;
        Arrays.fill(p, false);
        p[0] = true;
        int q = 0;
        int ss = sum/2;
        for (int d = 1; d < dig; d++) {
            for (int j = 0; j < a[d]; j++) {
                q++;
                for (int k = Math.min(sum/2 - d, q * (dig - 1)); k>=0; k--) p[k + d] |= p[k];
                
                if (p[ss]) break;
            }
            if (p[ss]) break;
        }
        return p[ss];
    }
 
    private void cnk() {
        c = new BigInteger[n + 1][n + 1];
        for (int i = 0; i <= n; i++) {
            c[i][0] = BigInteger.ONE;
            c[i][i] = BigInteger.ONE;
            for (int j = 1; j < i; j++) {
                c[i][j] = c[i - 1][j - 1].add(c[i - 1][j]);
            }
        }
    }
 
    public static void main(String[] args) throws IOException {
        //Locale.setDefault(Locale.US);
        new Thread(new Main()).start();
    }
 
    public void run() {
        try {
            in = new Scanner(new FileReader("input.txt"));
            out = new PrintWriter("output.txt");
            solve();
            in.close();
            out.close();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
            System.exit(1);#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
 
#define MAX_L 32
#define POW 100000
#define MAX_N 100
#define MAX_K 10
 
struct long_def
{
  int l;
  int a[MAX_L];
};
typedef struct long_def long_t;
 
FILE *fi, *fo;
int dig[MAX_K];
long_t ans_def;
long_t *ans;
int ar_1[MAX_N*MAX_K + 1];
int ar_2[MAX_N*MAX_K + 1];
int *a, *b, *t;
int N, K;
 
int min (int a, int b) { return (a<b)?a:b; }
int max (int a, int b) { return (a>b)?a:b; }
 
long_t *
zero (long_t *x)
{
  int i;
  x->l = 0;
  for (i = 0; i < MAX_L; i++)
    x->a[i] = 0;
  return x;
}
 
long_t *
one (long_t *x)
{
  zero (x);
  x->l = 1;
  x->a[0] = 1;
  return x;
}
 
 
long_t *
add_long (long_t *x, long_t *y, long_t *z)
{
  int i, r, l;
  l = max (x->l, y->l);
  r = 0;
  for (i = 0; i < l; i++)
    {
      z->a[i] = x->a[i] + y->a[i] + r;
      r = z->a[i] / POW;
      z->a[i] %= POW;
    }
  while (r > 0)
    {
      z->a[l++] = r % POW;
      r /= POW;
    }
  z->l = l;
  return z;
}
 
long_t *
mul_long (long_t *x, int q, long_t *z)
{
  int i, r, l;
  l = x->l;
  r = 0;
  for (i = 0; i < l; i++)
    {
      z->a[i] = x->a[i] * q + r;
      r = z->a[i] / POW;
      z->a[i] %= POW;
    }
  while (r > 0)
    {
      z->a[l++] = r % POW;
      r /= POW;
    }
  z->l = l;
  return z;
}
 
long_t *
div_long (long_t *x, int q, long_t *z)
{
  int i, r, l;
  l = x->l;
  r = 0;
  for (i = l - 1; i >= 0; i--)
    {
      z->a[i] = r * POW + x->a[i];
      r = z->a[i] % q;
      z->a[i] /= q;
    }
  while ((l > 0) && (z->a[l-1] == 0))
    l--;
  z->l = l;
  return z;
}
 
void
write_long (long_t *x)
{
  int i, l;
  l = x->l;
  fprintf (fo, "%d", x->a[l - 1]);
  for (i = l - 2; i >= 0; i--)
    {
      fprintf (fo, "%05d", x->a[i]);
    }
  fprintf (fo, "\n");
}
 
void
count (void)
{
  int i, j, d, n;
  int s;
  long_t tmp_def;
  long_t *tmp = &tmp_def;
  a = ar_1;
  b = ar_2;
  s = 0;
  for (i = 1; i <= K; i++)
    s += dig[i] * i;
  if (!(s%2))
    {
      for (i = 0; i <= s/2; i++)
    {
      a[i] = 0;
    }
      a[0] = 1;
      for (d = K; d > 0; d--)
    {
      for (i = 0; i <= s/2; i++)
        b[i] = a[i];
      for (i = 0; i <= s/2; i++)
        {
          if (!a[i]) continue;
          for (j = 0; j <= dig[d]; j++)
        {
          b[i + j*d] = 1;
        }
       }
      t = a; a = b; b = t;
    }
    }
  if ((!a[s/2]) || (s%2))
    {
      n = N;
      one (tmp);
      for (d = 0; d <= K; d++)
    {
      for (i = n - dig[d] + 1; i <= n; i++)
        mul_long (tmp, i, tmp);
      for (i = 2; i <= dig[d]; i++)
        div_long (tmp, i, tmp);
      n -= dig[d];
    }
      add_long (ans, tmp, ans);
    }
}
 
void
gen (int d, int s)
{
  int i;
  if (s > N) return;
  if (d > K)
    {
      dig[0] = N - s;
      count ();
      return ;
    }
  for (i = 0; i <= N - s; i++)
    {
      dig[d] = i;
      gen (d + 1, s + i);
    }
}
 
int
main (void)
{
  fi = fopen ("input.txt", "rt");
  fo = fopen ("output.txt", "wt");
 
  ans = &ans_def;
  while (1)
    {
      if (fscanf (fi, "%d%d", &N, &K) < 2) break;
      zero (ans);
      gen (1, 0);
      write_long (ans);
    }
 
  fclose (fi);
  fclose (fo);
  return 0;
}
        }
    }
}

Добавлено через 1 час 33 минуты
Появилась идея мгновенного счета, при k=1 ответ будет 2^(n-1)
при k=2 ответ будет a[0]=2;a[1]=6;a[2]=17;a[i] = a[i-1]*3-2^(i-1) - 1;
осталось выяснить при всех других

Добавлено через 38 минут
Даже еще эффективней
при k=1 ответ будет 2^(n-1)
при k=2 ответ будет 1/2*(-1+2^n+3^n)

Добавлено через 52 минуты
при k=3 ответ будет 1/2* (2^n*n-2*n+2^n+2^(2*n))

ValeryS · 03.02.2013, 12:46

словами объясни что нужно сделать
свой алгоритм словами
и кусок кода

@Mysterion777 · 03.02.2013, 12:53 **[ТС]**

Сообщение от ValeryS

словами объясни что нужно сделать
свой алгоритм словами
и кусок кода

Java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
import java.io.*;
import java.math.*;
import java.util.*;
 
public class Main implements Runnable {
    private Scanner in;
    private PrintWriter out;
    int dig = 10;
    
    int n;
    BigInteger[][] c;
    int[] a;
    boolean[] p;
    BigInteger answer;
    int parts;
 
    private void solve() {
        while (in.hasNextInt()){
          //  n = 2; 
         n=in.nextInt();dig=in.nextInt();   
            dig++;
            cnk();
            a = new int[dig];
            p = new boolean[(n+1) / 2 * (dig - 1) + 1];
            answer = BigInteger.ZERO;
            go(0,0,BigInteger.ONE,0);
            
       //    if(dig==5){
          //  System.out.print(answer+" ");
            out.println(answer);//}
        }
        
    }
 
    private void go(int i, int put, BigInteger amount, int sum) {
        if (i == dig) {
            if (put != n)
                return;
            if (sum % 2 != 0 ||  !happy(sum)) {
                answer = answer.add(amount);
            }
            return;
        }
        for (int m = 0; put + m <= n; m++) {
            a[i] = m;
            go(i + 1, put + m, amount.multiply(c[n - put][m]), sum + i * m);
        }
    }
 
    private boolean happy(int sum) {
        parts++;
        Arrays.fill(p, false);
        p[0] = true;
        int q = 0;
        int ss = sum/2;
        for (int d = 1; d < dig; d++) {
            for (int j = 0; j < a[d]; j++) {
                q++;
                for (int k = Math.min(sum/2 - d, q * (dig - 1)); k>=0; k--) p[k + d] |= p[k];
                
                if (p[ss]) break;
            }
            if (p[ss]) break;
        }
        return p[ss];
    }
 
    private void cnk() {
        c = new BigInteger[n + 1][n + 1];
        for (int i = 0; i <= n; i++) {
            c[i][0] = BigInteger.ONE;
            c[i][i] = BigInteger.ONE;
            for (int j = 1; j < i; j++) {
                c[i][j] = c[i - 1][j - 1].add(c[i - 1][j]);
            }
        }
    }
 
    public static void main(String[] args) throws IOException {
        //Locale.setDefault(Locale.US);
        new Thread(new Main()).start();
    }
 
    public void run() {
        try {
            in = new Scanner(new FileReader("input.txt"));
            out = new PrintWriter("output.txt");
          //  for(int j=1;j<40;j++)
          ///  for(int i=4;i<=4;i++){dig=i;n=j;
            solve();//}
            in.close();
            out.close();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
            System.exit(1);
        }
    }
}

ссылка на задачу ( http://acmp.ru/index.asp?main=task&id_task=435 )
Вот код генерирует все правильно и работает максимально оптимально....но должно укладываться в секунду. Я заметил что не обязательно так считать можно через формулу как я написал выше.Так вот например дана последовательность 4 20 94 424 1919 8874 42074 203600 998989 4943098 24576248 122509296 611551139 3055073690 15268067254 76320218080 381544943897 как определить формулу?

Новые блоги и статьи Все статьи Все блоги /
Изучаю kubernetes lagorue 13.01.2026 А пригодятся-ли мне знания kubernetes в России?	Сукцессия микоризы: основная теория в виде двух уравнений. anaschu 11.01.2026 https:/ / rutube. ru/ video/ 7a537f578d808e67a3c6fd818a44a5c4/	WordPad для Windows 11 Jel 10.01.2026 WordPad для Windows 11 — это приложение, которое восстанавливает классический текстовый редактор WordPad в операционной системе Windows 11. После того как Microsoft исключила WordPad из. . .	Classic Notepad for Windows 11 Jel 10.01.2026 Old Classic Notepad for Windows 11 Приложение для Windows 11, позволяющее пользователям вернуть классическую версию текстового редактора «Блокнот» из Windows 10. Программа предоставляет более. . .
Почему дизайн решает? Neotwalker 09.01.2026 В современном мире, где конкуренция за внимание потребителя достигла пика, дизайн становится мощным инструментом для успеха бренда. Это не просто красивый внешний вид продукта или сайта — это. . .	Модель микоризы: классовый агентный подход 3 anaschu 06.01.2026 aa0a7f55b50dd51c5ec569d2d10c54f6/ O1rJuneU_ls https:/ / vkvideo. ru/ video-115721503_456239114	Owen Logic: О недопустимости использования связки «аналоговый ПИД» + RegKZR ФедосеевПавел 06.01.2026 Owen Logic: О недопустимости использования связки «аналоговый ПИД» + RegKZR ВВЕДЕНИЕ Введу сокращения: аналоговый ПИД — ПИД регулятор с управляющим выходом в виде числа в диапазоне от 0% до. . .	Модель микоризы: классовый агентный подход 2 anaschu 06.01.2026 репозиторий https:/ / github. com/ shumilovas/ fungi ветка по-частям. коммит Create переделка под биомассу. txt вход sc, но sm считается внутри мицелия. кстати, обьем тоже должен там считаться. . . .

ValeryS Модератор 8978 / 6744 / 921 Регистрация: 14.02.2011 Сообщений: 23,853
	03.02.2013, 12:46
	словами объясни что нужно сделать свой алгоритм словами и кусок кода 0