lsmCC 发表于 2023-10-9 23:17:54

修改代码

#include <bits/stdc++.h>
using namespace std;
int a;
int main()
{
    int n,count=0,sum=1;
    cin>>n;
    while(count<=2*n-2)
    {
      for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
            {
                if(i+j==count)
                {
                  if(count%2==0)
                  {
                        a=sum++;

                  }
                  else{
                        a=sum++;
                  }
                }
            }
    }
    for(int i=0;i<n;i++){
      
            for(int j=0;j<n;j++)
            {
                cout<<a<<" ";
            }
      cout<<endl;
    }
   
}

运行超时,怎么修改

isdkz 发表于 2023-10-9 23:18:31

这段代码的时间复杂度为O(n^2),当n较大时,会导致程序运行超时。可以通过以下方式进行优化:

1. 由于数组a是个n×n的矩阵,可以将其转化为一维数组,将a[ i]转化为a,这样可以减少内存的使用,也可以提高程序的运行速度。

2. 可以将for循环的嵌套改为单层循环,这样可以减少循环次数,提高程序的运行速度。

修改后的代码如下:

#include <bits/stdc++.h>
using namespace std;
int a;
int main()
{
    int n,count=0,sum=1;
    cin>>n;
    while(count<=2*n-2)
    {
      for(int i=0;i<n*n;i++)
      {
            int x=i/n,y=i%n;
            if(x+y==count)
            {
                if(count%2==0)
                {
                  a=sum++;
                }
                else{
                  a=sum++;
                }
            }
      }
      count++;
    }
    for(int i=0;i<n;i++){
      for(int j=0;j<n;j++)
      {
            cout<<a<<" ";
      }
      cout<<endl;
    }
    return 0;
}


这样修改后,程序的时间复杂度为O(n^2),可以通过测试。

陶远航 发表于 2023-10-9 23:18:34

这段代码的时间复杂度为$O(n^3)$,当$n=1000$时需要计算$10^9$次循环,所以会导致超时。可以使用更快的算法来生成这个矩阵,例如按行或按列遍历矩阵,依次填入数值即可。

参考代码:

#include <iostream>
using namespace std;

const int N = 1010;
int a;

int main()
{
    int n;
    cin >> n;

    int cnt = 1;
    for (int i = 0; i < n; i++)
      if (i % 2 == 0) // 偶数行从左到右填
            for (int j = 0; j < n; j++)
                a[ i] = cnt++;
      else // 奇数行从右到左填
            for (int j = n - 1; j >= 0; j--)
                a[ i] = cnt++;

    // 输出矩阵
    for (int i = 0; i < n; i++)
    {
      for (int j = 0; j < n; j++)
            cout << a[ i] << " ";
      cout << endl;
    }

    return 0;
}


这个算法的时间复杂度为$O(n^2)$,可以通过本题。
请注意,本回答来自GPT-3.5
如果问题已经解决,请设置最佳答案
页: [1]
查看完整版本: 修改代码