可能由于模（10,9）+7 导致超出时间限制错误？

Question

我们的 X 轴上有 N 个士兵。士兵所在的位置也有一些炸弹。 战争临近，每个士兵都想与其他士兵交流。 如果i-th士兵有b个炸弹并且位于位置X，那么与任何其他在位置Y有c个炸弹的士兵j通信的成本被定义为|X-Y|*max(b,c)。 如果每个士兵都想与其他士兵交流，求交流成本的总和。 注意：- 您必须在总成本中只考虑一次 pair(i,j)。

输入格式：

第一行包含测试用例的数量 T。每个测试用例包含三行。第一行表示士兵人数（N）。第二行表示 N 个士兵的坐标（ X[i] ）。第三行包含每个士兵位置的炸弹数量 (B[i])。 x 坐标在输入中不必按升序排列。 约束

1 <= T <= 20 1 <= N <= 200000 1 <= X[i] <= 1000000000 1 <= B[i] <= 10000

输出格式：

总成本模 10^9+7。 样本输入

1
3  
1 3 6  
10 20 30  

样本输出

280

解释

有 3 对 (1,2) -> 成本 = abs(3-1) * 20 = 40 (1,3) -> 成本 = abs(1-6) * 30 = 150 (2,3) -> 成本 = abs(3-6) * 30 = 90 总和 = 40 + 150 + 90 = 280

我正在处理模数 (10^9+7) 和一切使用蛮力但得到低于其的 tle 代码也适用于上述情况但它是那些烦人的 tle/类型转换类型的问题之一。任何回应都非常感谢-

import java.util.*;
public class Main {
public static void main(String args[]) {
    Scanner sc= new Scanner(System.in);
    int T=sc.nextInt();

    for(int ctr=0;ctr<T;ctr++)
    {
        int N=sc.nextInt();
        long [] x= new long[N];
        long [] b= new long[N];
        for(int i=0;i<N;i++)
        {
            x[i]=sc.nextLong();

        }
        for(int i=0;i<N;i++)
        {
            b[i]=sc.nextInt();

        }
        int cost=0;
        double v;
        for(int i=0;i<N-1;i++)
        {
            for(int j=i+1;j<N;j++)
            {
                v= x[i]%(Math.pow(10,9)+7)-x[j]%(Math.pow(10,9)+7);
                v=Math.abs(v)%(Math.pow(10,9)+7);
                cost+= v*Math.max(b[i],b[j]);
            }
        }
        System.out.println(cost);

    }
}   }

Answer 1

正如@Erwin 所建议的，不要在for 循环的每次迭代中使用Math.pow() 来计算(10^9)+7 三次。相反，将其置于循环之外（也许将其定义为常量），然后执行您的操作。

此外，您在运算中对 (10^9)+7 常量进行两次模运算。你确定这是预期的操作吗？您在v 的第一次计算中执行此操作，然后在v 的第二次计算中再次执行此操作。如果您只需要它来支付最终费用，您可以从第一步中删除它。

这是我的代码，你可以试试：

import java.util.*;
public class Main {
public static void main(String args[]) {
    double moduloNumber = (Math.pow(10,9)+7);
    Scanner sc= new Scanner(System.in);
    System.out.print("Enter the Number of Test Cases (1 <= T <= 20): ");
    int T=sc.nextInt();

    for(int ctr=0;ctr<T;ctr++)
    {
        System.out.print("\nEnter the Number of Soldiers (1 <= N <= 200000): ");
        int N=sc.nextInt();
        long [] x= new long[N];
        long [] b= new long[N];
        for(int i=0;i<N;i++)
        {
            System.out.print("Enter the Position of Soldier " + (i+1) + " (1 <= X[i] <= 1000000000): ");
            x[i]=sc.nextLong();

        }
        for(int i=0;i<N;i++)
        {
            System.out.print("Enter the Number of Bombs with Soldier " + (i+1) + " (1 <= B[i] <= 10000): ");
            b[i]=sc.nextInt();

        }
        int cost=0;
        double v;
        for(int i=0;i<N-1;i++)
        {
            for(int j=i+1;j<N;j++)
            {
                v=(x[i]-x[j]);
                v=Math.abs(v)%(moduloNumber);
                cost+= v*Math.max(b[i],b[j]);
            }
        }
        System.out.println(cost);

    }
}
}

我只是添加了一些打印语句并将模常数移到循环之外以降低计算成本。您可以检查这是否解决了目的。

问候，

AJ