Time limit exceed problem

[chonchon430]

I got the following question in my exercise:

"There is 2 integers, x and y and I need to classify the integer from 1 to y into 2 group. One group with integers that are multiple of y and the other is the remaining integer. Afterward, I need to calculate the mean value of each group."

I have written the following code and the direction should be correct, but there will be a time limit exceed problem when the size of data become very large.

x,y = [int(num) for num in input().split()]
A=[]
B=[]
for i in range(1,x+1):
    if i%y==0:
        A.append(i)
    else:
        B.append(i)
print(round((sum(A)/len(A)),1),round((sum(B)/len(B)),1))

I hope to find some suggestion to solve the problem, either modify my current code or give a new direction for me to write the code will be find. Thank you very much!

[DPaul]

Maybe an example input and the desired output would help.

Paul

[chonchon430]

Maybe an example input and the desired output would help.

Paul

example input : 100 16
desired output: 56.0 50.1

[DPaul]

"There is 2 integers, x and y and I need to classify the integer from 1 to y into 2 group. One group with integers that are multiple of y and the other is the remaining integer. Afterward, I need to calculate the mean value of each group."

Is this description of the problem 100% accurate ?
Does 1 need to be x ?
Does the last "integer" need to be "integers"?
It makes a big difference.
Paul

[chonchon430]

"There is 2 integers, x and y and I need to classify the integer from 1 to y into 2 group. One group with integers that are multiple of y and the other is the remaining integer. Afterward, I need to calculate the mean value of each group."

Is this description of the problem 100% accurate ?
Does 1 need to be x ?
Does the last "integer" need to be "integers"?
It makes a big difference.
Paul

Opps...sorry I have make a mistake. The description should be:
"There is 2 integers, x and y and I need to classify the integer from 1 to x into 2 group. One group with integers that are multiple of y and the other is the remaining integer. Afterward, I need to calculate the mean value of each group."

For example when the sample input is 100 16, I first need to separate the integer from 1 to 100 into 2 group. The first group is the integer that is multiple of 16 in 1 to 100 and the second group is the integers that is not the multiple of 16(i.e. the integer from 1 to 100 that is not in the first group.) Afterward, I need to calculate the mean of the two group.

I hope this will make you clearer, thanks!

[DPaul]

There is an input ? on line 1, that i made simpler

x,y = 100,16

I've written about the same program that you did,
and ran it with x = 10_000, no problem, instantaneous.
What would be your definition of "very large?"
Paul

[chonchon430]

There is an input ? on line 1, that i made simpler

x,y = 100,16

I've written about the same program that you did,
and ran it with x = 10_000, no problem, instantaneous.
What would be your definition of "very large?"
Paul

Yup, it is ok when the program is run in idle. However when putting into the online judge system, which has a time limit of 1000MS and memory limit of 256MB, the TLE problem will appear. Thus, I am seeking way to further simplify my code.....

[DPaul]

OK, if that is your problem, maybe a suggestion:
You can make your vector of multiples of 16 , directly

[16 * i for i in range(1,int(100/16 + 1))]

You can do the same for 1 to 100.
Then with list comprehension make a new list 1-100 that does not contain the multiples of 16.

Whether that is faster, i have not timed. You are saving a lot of unnecessary divisions.
Probably there is plenty of room for other ideas
Paul

[chonchon430]

OK, if that is your problem, maybe a suggestion:
You can make your vector of multiples of 16 , directly

[16 * i for i in range(1,int(100/16 + 1))]

You can do the same for 1 to 100.
Then with list comprehension make a new list 1-100 that does not contain the multiples of 16.

Whether that is faster, i have not timed. You are saving a lot of unnecessary divisions.
Probably there is plenty of room for other ideas
Paul

Thanks for your suggestion!
I successfully make the multiple of 16 directly in a list, however I encounter problem in the list does not contain the multiples of 16...

For example A=[list of multiple of 16], and I make the other list with the following list:

B=[j for j in range(1,100+1) if j!=[A]]

The list return still contain the multiple of 16... May I know what's wrong with my code. Thanks!

[DPaul]

B=[j for j in range(1,100+1) if j!=[A]]

[A] is not what you need.A is a vector in its own right.
Try this:

.... if not in A.....

Paul