首页 > 代码库 > 用awk写递归

用awk写递归

看到自己很多年前写的一篇帖子,觉得有些意义,转录过来,稍加修改。

awk是一种脚本语言,语法接近C语言,我比较喜欢用,gawk甚至可以支持tcp/ip,用起来非常方便。

awk也支持递归,只是awk不支持局部变量,所有的变量都是全局的,于是写递归有些麻烦。本文说白了,也只是借awk说一种编程的思路罢了。

原文如下:

awk支持函数,也支持递归。但awk并不支持局部变量,于是看上去递归函数很不好实现,因为在某一级调用函数的时候,里面的变量在该级调用还没有退出前就可能会被别的调用给修改掉,于是得到的结果会与期望并不一致。
我们考虑C语言,它的局部变量放在硬件支持的栈(一般用栈指针)内。于是我们就去思考,为什么是栈呢?我们来考虑一个具体的函数调用顺序:
f1调用f2;
f2调用f3;
f3返回;
f2调用f4;
f4调用f5;
f5返回;
f4返回;
f2返回;
f1返回;
按照这个循序,我们来思考每个函数开辟的栈空间:
f1的栈空间开辟(f1进栈)
f2的栈空间开辟(f2进栈)
f3的栈空间开辟(f3进栈)
f3的栈空间消亡(f3出栈)
f4的栈空间开辟(f4进栈)
f5的栈空间开辟(f5进栈)
f5的栈空间消亡(f5出栈)
f4的栈空间消亡(f4出栈)
f2的栈空间消亡(f2出栈)
f1的栈空间消亡(f1出栈)
原来跟我们数据结构里的栈的先进后出是一回事情,所以叫栈。
而当前的我们取的变量的地址都是相对于栈指针来说的,这是相对,而不是像全局变量的那种绝对。
于是我们可以受到启发,可以扩展这里的栈指针和地址的概念,awk的递归函数就可以出来了。
以下是用递归来算一个数组中的最大值(每递归一级就把数组分为两段,每段求最大值),只是举一个例子,可以扩展到任意应用。

#!/bin/awk -ffunc test1(a,start,len){        if(len<=1)                return a[start];        x = test1(a,start,int(len/2));        y = test1(a,start+int(len/2),len-int(len/2));        return (x>y)?x:y;}func test2(a,start,len){        if(len<=1)                return a[start];        testlen++;        testa[testlen] = test2(a,start,int(len/2));        testlen++;        testa[testlen] = test2(a,start+int(len/2),len-int(len/2));        testlen-=2;        return (testa[testlen+1]>testa[testlen+2])?testa[testlen+1]:testa[testlen+2];}func test3(a,start,len){        if(len<=1) {                return a[start];        }        V = V";"test3(a,start,int(len/2));        V = V";"test3(a,start+int(len/2),len-int(len/2));        xx = V;        sub(/.*;/,"",xx);        sub(/;[^;]+$/,"",V);        yy = V;        sub(/.*;/,"",yy);        sub(/;[^;]+$/,"",V);        return int(xx)>int(yy)?int(xx):int(yy);}NR==1{        way=$1;        print $1}NR==2{        max=$1;        for(i=2;i<=NF;i++)                if($i > max)                        max = $i;        print max;        for(i=1;i<=NF;i++)                a[i] = $i;        if(way == 2)                print test2(a,1,NF);        else if(way == 3)                print test3(a,1,NF);        else                print test1(a,1,NF);        exit(0);}

这里面实现了三个递归函数,第一个是测试全局变量的污染,它是得不到正确的答案的
第二个是用数组来模拟变量栈,testlen就是所谓的“栈顶指针”
第三个是用字符串来模拟变量栈,字符串末尾就是“栈顶指针”,每个“局部变量”之间是用分号隔开
用随机数据测试一下这个应用:

linux-0gt0:/tmp/test # for((i=0;i<10;i++));do  { echo $(($RANDOM % 3 + 1)); let count=$RANDOM%100+50; for((j=0;j<count;j++));do echo -n $(($RANDOM % 10000)) " "; done ; echo ; }|./1.awk ;done | sed -nr ‘N;N;p;/\n(.*)\n\1$/{s/.*/right\n/p;d;};       /^[23]/s/(.).*/\1 SO STRANGE!!!!!!!!!!!!!!!!!!!!!/p; s/.*/wrong\n/p‘299819981right393919391right199195257wrong298609860right399679967right399409940right398289828right297529752right399969996right299309930right

当然,栈的数目自然也可以不只维系一个,test2和test3维系的是一个栈。
现在来实现test4和test5,两个函数是test2和test3的变体,各自维系两个栈。

#!/bin/awk -f#func test1(a,start,len)#{#       if(len<=1)#               return a[start];#       x = test1(a,start,int(len/2));#       y = test1(a,start+int(len/2),len-int(len/2));#       return (x>y)?x:y;#}func test2(a,start,len){        if(len<=1)                return a[start];        testlen++;        testa[testlen] = test2(a,start,int(len/2));        testlen++;        testa[testlen] = test2(a,start+int(len/2),len-int(len/2));        testlen-=2;        return (testa[testlen+1]>testa[testlen+2])?testa[testlen+1]:testa[testlen+2];}func test3(a,start,len){        if(len<=1) {                return a[start];        }        V = V";"test3(a,start,int(len/2));        V = V";"test3(a,start+int(len/2),len-int(len/2));        xx = V;        sub(/.*;/,"",xx);        sub(/;[^;]+$/,"",V);        yy = V;        sub(/.*;/,"",yy);        sub(/;[^;]+$/,"",V);        return int(xx)>int(yy)?int(xx):int(yy);}func test4(a,start,len){        if(len<=1)                return a[start];        testlenx++;        testleny++;        testax[testlenx] = test4(a,start,int(len/2));        testay[testleny] = test4(a,start+int(len/2),len-int(len/2));        testlenx-=1;        testleny-=1;        return (testax[testlenx+1]>testay[testleny+1])?testax[testlenx+1]:testay[testleny+1];}func test5(a,start,len){        if(len<=1) {                return a[start];        }        V1 = V1";"test5(a,start,int(len/2));        V2 = V2";"test5(a,start+int(len/2),len-int(len/2));        xx = V1;        sub(/.*;/,"",xx);        sub(/;[^;]+$/,"",V1);        yy = V2;        sub(/.*;/,"",yy);        sub(/;[^;]+$/,"",V2);        return int(xx)>int(yy)?int(xx):int(yy);}NR==1{        way=$1;        print $1}NR==2{        max=$1;        for(i=2;i<=NF;i++)                if($i > max)                        max = $i;        print max;        for(i=1;i<=NF;i++)                a[i] = $i;        if(way == 2)                print test2(a,1,NF);        else if(way == 3)                print test3(a,1,NF);        else if(way == 4)                print test4(a,1,NF);        else if(way == 5)                print test5(a,1,NF);        exit(0);}

测试一下,

linux-0gt0:/tmp/test # for((i=0;i<10;i++));do  { echo $(($RANDOM % 2 + 4)); let count=$RANDOM%100+50; for((j=0;j<count;j++));do echo -n $(($RANDOM % 10000)) " "; done ; echo ; }|./1.awk ;done | sed -nr ‘N;N;p;/\n(.*)\n\1$/{s/.*/right\n/p;d;};       /^[234]/s/(.).*/\1 SO STRANGE!!!!!!!!!!!!!!!!!!!!!/p; s/.*/wrong\n/p‘599049904right498239823right599759975right499669966right596839683right599819981right499839983right599669966right599679967right598709870right

当然,test4和test5各自维系的两个栈其实差别和一个栈不大,因为两个栈是同时进栈同时出栈的。
其实,即使两个栈并非同时进出栈也是可以的,只是对于这里的例子来说写不出这么复杂。
实际上,任意多的栈,任意进出栈,都是可以的。
这样就可以做到更加灵活的应用。
软件的扩展性比硬件强,我想,这就是软件的用处吧。

还是这个取最大值,举个含有多个栈,每个栈入出并不完全一致的例子,这里的test6函数

#!/bin/awk -ffunc test1(a,start,len){        if(len<=1)                return a[start];        x = test1(a,start,int(len/2));        y = test1(a,start+int(len/2),len-int(len/2));        return (x>y)?x:y;}func test2(a,start,len){        if(len<=1)                return a[start];        testlen++;        testa[testlen] = test2(a,start,int(len/2));        testlen++;        testa[testlen] = test2(a,start+int(len/2),len-int(len/2));        testlen-=2;        return (testa[testlen+1]>testa[testlen+2])?testa[testlen+1]:testa[testlen+2];}func test3(a,start,len){        if(len<=1) {                return a[start];        }        V = V";"test3(a,start,int(len/2));        V = V";"test3(a,start+int(len/2),len-int(len/2));        xx = V;        sub(/.*;/,"",xx);        sub(/;[^;]+$/,"",V);        yy = V;        sub(/.*;/,"",yy);        sub(/;[^;]+$/,"",V);        return int(xx)>int(yy)?int(xx):int(yy);}func test4(a,start,len){        if(len<=1)                return a[start];        testlenx++;        testleny++;        testax[testlenx] = test4(a,start,int(len/2));        testay[testleny] = test4(a,start+int(len/2),len-int(len/2));        testlenx-=1;        testleny-=1;        return (testax[testlenx+1]>testay[testleny+1])?testax[testlenx+1]:testay[testleny+1];}func test5(a,start,len){        if(len<=1) {                return a[start];        }        V1 = V1";"test5(a,start,int(len/2));        V2 = V2";"test5(a,start+int(len/2),len-int(len/2));        xx = V1;        sub(/.*;/,"",xx);        sub(/;[^;]+$/,"",V1);        yy = V2;        sub(/.*;/,"",yy);        sub(/;[^;]+$/,"",V2);        return int(xx)>int(yy)?int(xx):int(yy);}func test6(a,start,len){        if(len <= 1) {                return a[start];        } else if(len == 2) {                return (a[start]>a[start+1])?a[start]:a[start+1];        } else if(len == 3) {                var1 = (a[start]>a[start+1])?a[start]:a[start+1];                return (var1>a[start+2])?var1:a[start+2];        }        var2 = int(rand()*10000)%3+2;        if(var2 == 2) {                testlenx++;                testleny++;                testax[testlenx] = test6(a,start,int(len/2));                testay[testleny] = test6(a,start+int(len/2),len-int(len/2));                testlenx-=1;                testleny-=1;                return (testax[testlenx+1]>testay[testleny+1])?testax[testlenx+1]:testay[testleny+1];        } else if(var2 == 3) {                testlenx++;                testleny++;                testlenz++;                testax[testlenx] = test6(a,start,int(len/3));                testay[testleny] = test6(a,start+int(len/3),int(len/3));                testaz[testlenz] = test6(a,start+2*int(len/3),len-2*int(len/3));                testlenx-=1;                testleny-=1;                testlenz-=1;                var1 = (testax[testlenx+1]>testay[testleny+1])?testax[testlenx+1]:testay[testleny+1];                return ((var1>testaz[testlenz+1])?var1:testaz[testlenz+1]);        } else if(var2 == 4) {                testlenx++;                testleny++;                testlenz++;                testlenA++;                testax[testlenx] = test6(a,start,int(len/4));                testay[testleny] = test6(a,start+int(len/4),int(len/4));                testaz[testlenz] = test6(a,start+2*int(len/4),int(len/4));                testaA[testlenA] = test6(a,start+3*int(len/4),len-3*int(len/4));                testlenx-=1;                testleny-=1;                testlenz-=1;                testlenA-=1;                var1 = (testax[testlenx+1]>testay[testleny+1])?testax[testlenx+1]:testay[testleny+1];                var4 = (testaz[testlenz+1]>testaA[testlenA+1])?testaz[testlenz+1]:testaA[testlenA+1];                return ((var1>var4)?var1:var4);        }}BEGIN {        srand(systime());}NR==1{        way=$1;        print $1}NR==2{        max=$1;        for(i=2;i<=NF;i++)                if($i > max)                        max = $i;        print max;        for(i=1;i<=NF;i++)                a[i] = $i;        if(way == 2)                print test2(a,1,NF);        else if(way == 3)                print test3(a,1,NF);        else if(way == 4)                print test4(a,1,NF);        else if(way == 5)                print test5(a,1,NF);        else if(way == 6)                print test6(a,1,NF);        else                print test1(a,1,NF);        exit(0);}

  

用awk写递归