C语言循环小技巧

写代码,有两类追求,一种是追求实用(Coder),一种是追求代码艺术(Artist)
我是那种追实用追腻了,偶然追一下艺术(就是偶然和艺术有一腿)的那种Coder


很多人,已经习惯了for(i=0; i<n; i++)这种单调的循环,虽然这的确的使用率最高,
但在特殊场合,特殊的循环写法,不但能提升循环的效率,还能使代码更精巧

1. 质数判断
对于这个,很多人可能会直接这样写:
int isPrime(int n) //函数返回1表示是质数,返回0表示不是质数
{
    int i;
    for (i = 2; i < n; i++)
        if (n % i == 0)
            break;
    return i >= n;
}

又或者,有的人知道平方根的优化:
int isPrime(int n)
{
    int i, s = (int)(sqrt((double)n) + 0.01);
    for (i = 2; i <= s; i++)
        if (n % i == 0)
            break;
    return i > s;
}
再或者,消除偶数:
int isPrime(int n)
{
    int i, s = (int)(sqrt((double)n) + 0.01);
    if (n <= 3) return 1;
    if (n % 2 == 0) return 0;
    for (i = 3; i <= s; i += 2)
        if (n % i == 0)
            break;
    return i > s;
}

当然,这样还不是很够的话,我们可以考虑这个事实:
所有大于4的质数,被6除的余数只能是1或者5
比如接下来的5,7,11,13,17,19都满足

所以,我们可以特殊化先判断2和3
但后面的问题就出现了,因为并非简单的递增,从5开始是+2,+4,+2,+4,....这样递增的
这样的话,循环应该怎么写呢?

首先,我们定义一个步长变量step,循环大概是这样 for (i = 5; i <= s; i += step)
那么,就是每次循环,让step从2变4,或者从4变2
于是,可以这么写:

#include <stdio.h>
#include <math.h>

int isPrime(int n)
{
    int i, s = (int)(sqrt((double)n) + 0.01), step = 4;
    if (n <= 3) return 1;
    if (n % 2 == 0) return 0;
    if (n % 3 == 0) return 0;
    for (i = 5; i <= s; i += step)
    {
        if (n % i == 0)
            break;
        step ^= 6;
    }
    return i > s;
}

int main()
{
    int n;
    for (n = 2; n < 100; ++n) //找出 2 - 100 的质数并输出
    {
        if (isPrime(n)) printf("%d,", n);
    }
    getchar();
    return 0;
}

如上代码,一个 step ^= 6; 完成step在2和4之间转换(这个 ^ 符号是C里的异或运算)
理由是,2化二进制是010,4是100,6是110,于是2异或4得到6:
2 ^ 4 => 6
6 ^ 2 => 4
6 ^ 4 => 2

于是利用异或,就可以构造这种步长在两个值之间来回变化的循环
思考题:前面说的是双值循环,那么如何构造三值或者四值循环?








2.菱形打印

很多人,打印菱形在控制台的思路是,把菱形上下拆分,分两段很接近的代码来打印,
其实这样代码很不好看,并且不好阅读
我们知道,要打印的图案是这种:
    *
   ***
  *****
   ***
    *

满足上下对称,左右对称,那么,你能不能也弄一个二重循环,同样是对称的?
很简单,首先我们要抛开习惯性思维,for循环不一定要在0开始或者0结束
我们可以让循环从 -c 到 c ,这样不就轻松产生一个对称的吗?(只要取个绝对值)
我们把菱形的中心看成是坐标0,0,那么,会输出星号的坐标,是 |x| + |y| <= c 的点

由此可得
#include <stdio.h>
#define IABS(x) ( (x) >= 0 ? (x) : -(x) ) //定义一个计算绝对值的宏
void print(int size) // size是这个菱形的半径,直径会是size * 2 + 1
{
    int x, y;
    for (y = -size; y <= size; y++)
    {
        for (x = -size; x <= size; x++)
        {
            if ( IABS(x) + IABS(y) <= size ) //x和y各自的绝对值的和,即 |x| + |y| <= size
                putchar('*');
            else
                putchar(' ');
        }
        putchar('\n');
    }
}

int main()
{
    print(5); //输出一个半径为5的菱形
    getchar();
    return 0;
}

如果我需要得到空心菱形呢?非常非常简单,因为菱形边界上的点,满足的是|x| + |y| == c
所以,我们只要把那个if里的小于等于号,改成双等于号 == 就可以了

再类似地,如果我不要*号,我要最外层是字母A,然后里一层是B这样呢?即:
    A
   ABA
  ABCBA
   ABA
    A

那么,我们只要在putchar那里做一个字符计算:
void print(int size) // size是这个菱形的半径,直径会是size * 2 + 1
{
    int x, y;
    for (y = -size; y <= size; y++)
    {
        for (x = -size; x <= size; x++)
        {
            if ( IABS(x) + IABS(y) <= size ) //x和y各自的绝对值的和,即 |x| + |y| <= size
                putchar( 'A' + (size - IABS(x) - IABS(y)) ); //留意这里的计算方法
            else
                putchar(' ');
        }
        putchar('\n');
    }
}

类似地,如果我们要打印的是X形:
  *   *
   * *
    *
   * *
  *   *
同样可以利用这个思路完成,这题就作为思考题吧








3. 奇数阶幻方
所谓幻方(最基本的那种),就是横,竖,对角线上的数的和等于一个常数的数字方阵
4 3 8
9 5 1
2 7 6

以上这个图,有什么规律?容易写成代码吗?

我们把这个图,向右复制五次,向下复制三次,展开一下:

 4  3  8  4  3  8  4  3  8  4  3  8  4  3  8
 9  5 [1] 9  5  1  9  5  1  9  5  1  9  5  1
 2  7  6 [2] 7  6  2  7  6  2  7  6  2  7  6
 4  3  8  4 [3] 8 [4] 3  8  4  3  8  4  3  8
 9  5  1  9  5  1  9 [5] 1  9  5  1  9  5  1
 2  7  6  2  7  6  2  7 [6] 2 [7] 6  2  7  6
 4  3  8  4  3  8  4  3  8  4  3 [8] 4  3  8
 9  5  1  9  5  1  9  5  1  9  5  1 [9] 5  1
 2  7  6  2  7  6  2  7  6  2  7  6  2  7  6

注意中括号数字的走向
怎么样,现在呢?
现在看起来显得规律性强了很多,但是,你会不会觉得循环还是不太好写?
我们如何从一个给定的n,直接得知它的坐标呢?
不难,找一下规律就可以发现对于任意的数值n+1有(以左上角为0,0坐标):
x = 2 + n + n / 3;
y = 1 + n - n / 3;

其实这个规律可以简单扩展到任意奇数阶幻方(以下size是奇数):
x = size / 2 + 1 + n + n / size; (注意这里的除法是取整除法,不带小数)
y = size / 2     + n - n / size;

这样,我们就可以把原来复杂的循环,化简成一重简单循环

于是有程序:
#include <stdio.h>
#define SIZE 5  //定义幻方阶数,这个数只能是奇数
int main()
{
    int x, y, i, sqSize, hSize;
    int sqMap[SIZE][SIZE];
    sqSize = SIZE * SIZE;
    hSize  = SIZE / 2;
    //计算1至SIZE * SIZE的数的位置并记录
    for ( i = 0; i < sqSize; i++)
    {
        x = hSize + 1 + i + i / SIZE;
        y = hSize     + i - i / SIZE;
        sqMap[y % SIZE][x % SIZE] = i + 1;
    }
    //以下是输出
    for (y = 0; y < SIZE; y++)
    {
        for (x = 0; x < SIZE; x++)
            printf("%4d", sqMap[y][x]);
        puts("");
    }
    return 0;
}


这个比你网上能找到的很多求奇数阶幻方的代码都短小很多(不过网上较多称之为魔方阵,不知为何)








4. 字符串循环移位

问题,给你一个字符串,要求循环左移n位
比如对"abcdefg" 循环左移2位,我们要得到"cdefgab"

附加条件,不能使用连续辅助空间(包括动态分配),只能使用若干单个变量(即O(1)空间)

首先,我们知道,反转一个字符串操作("abcd"变"dcba"),是不需要额外数组辅助的,只要头尾数据交换就可以了
然而,可能你不知道,仅仅使用字符串反转可以实现字符串循环移位:

//反转字符串,把st与ed所指向的中间的内容反转(包含st不包含ed)
void str_rev(char* st, char *ed)
{
    for (--ed; st < ed; ++st, --ed)
    {
        char c;
        c = *st; *st = *ed; *ed = c;
    }
}

//用三反转等效左移字符串(st与ed之间,包含st不包含ed的内容)
char* str_shl(char* st, char* ed, int n)
{
    str_rev(st, &st[n]);
    str_rev(    &st[n], ed);
    str_rev(st,         ed);
    return st;
}

#include <stdio.h>
#include <string.h>
int main()
{
    char str[] = "abcdefghijklmnopqrstuvwxyz";
    puts( str_shl(str, str + strlen(str), 6) );
    getchar();
    return 0;
}

这里,如果要循环左移n位,只要把原来字符串分成两段,前n字符,和后面其它字符
两段分别反转,最后再整体反转,就实现了循环左移(如果先整体再两部分,就是循环右移)

而在那个字符串反转函数里,参与循环的,不再是int,而是两个指针,
为什么选择使用两个指针呢?如果你写一个str_rev(char* str, int len)的版本,相信你就明白了,这里不多废话
原文地址:https://www.cnblogs.com/pianist/p/3286145.html