news 2026/4/3 3:02:45

dlx求解数独duckdb插件的编写和使用

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
dlx求解数独duckdb插件的编写和使用

1.将网上下载的dlx求解c程序添加int sudoku(const char *s,char *r)函数处理81个字符长的数独题目字符串

#include<cstdio>#include<cstring>#include<ctime>intcnt=0;constintXSIZE=3;constintSIZE=XSIZE*XSIZE;constintMAX_C=SIZE*SIZE*4;//最大列constintMAX_R=SIZE*SIZE*SIZE;//最大行constintMAX_SUDOKU=SIZE*SIZE;//数独矩阵大小constintMAX_LINK=MAX_C*MAX_R;//链表最大范围intL[MAX_LINK],R[MAX_LINK],U[MAX_LINK],D[MAX_LINK];//抽象链表intC[MAX_LINK],O[MAX_LINK],S[MAX_C],H[MAX_R];//C&O代表列&行,S每一列的节点数,H每一行的第一个节点intNodeNumber,RecordNumber;//用来指向节点intstate[MAX_SUDOKU],ans[MAX_SUDOKU],record[MAX_SUDOKU];//////////////////////Dancing Links模版//////////////////////voidinit(void);//Dancing Links的抽象链表初始化voidinsert(int,int);//在链表的一个位置中添加标记voidremove(int);//删除一列,同时删除这一列中的行voidresume(int);//恢复一列,同时恢复这一列中的行//////////////////////Dancing Links模版//////////////////////boolinput(void){chars[82]="";if(scanf("%s",s)==EOF)returnfalse;printf("Q %s\n",s);for(inti=0;i<81;i++)state[i]=s[i]-'0';returntrue;/* char buffer[SIZE+1][SIZE+1];//留一个空间 if(scanf("%s",buffer[0])==EOF) return false; for(int i=1;i<SIZE;i++) scanf("%s",buffer[i]); memset(state,0,sizeof(state)); for(int i=0;i<SIZE;i++) for(int j=0;j<SIZE;j++) { if(buffer[i][j]!='0') state[i*SIZE+j]=buffer[i][j]-'0'; } return true; */}voidadd(inti,intj,intk){introw=i*MAX_SUDOKU+j*SIZE+k;insert(row,i*SIZE+j+1);//注意insert(row,i*SIZE+k+MAX_SUDOKU);insert(row,j*SIZE+MAX_SUDOKU*2+k);insert(row,(i/XSIZE*XSIZE+j/XSIZE)*SIZE+MAX_SUDOKU*3+k);}voidbuild(void){intpos,row;for(inti=0;i<SIZE;i++)for(intj=0;j<SIZE;j++){pos=i*SIZE+j;if(state[pos]!=0){add(i,j,state[pos]);}elseif(state[pos]==0){for(intk=1;k<=SIZE;k++){add(i,j,k);}}}}booldfs(intk){cnt++;if(!R[0]){RecordNumber=k;returntrue;}intcount=~(1<<31),c;for(inti=R[0];i;i=R[i]){if(S[i]<count){count=S[i];c=i;if(count==1)break;}}remove(c);for(inti=D[c];i!=c;i=D[i]){for(intj=R[i];j!=i;j=R[j]){remove(C[j]);}record[k]=O[i];if(dfs(k+1))returntrue;for(intj=L[i];j!=i;j=L[j]){resume(C[j]);}}resume(c);returnfalse;}voidoutput(void){for(inti=0;i<RecordNumber;i++){ans[(record[i]-1)/SIZE]=(record[i]-1)%SIZE+1;}for(inti=0;i<SIZE;i++){for(intj=0;j<SIZE;j++)printf("%-2d",ans[i*SIZE+j]);printf("\n");}}intsudoku(constchar*s,char*r){for(inti=0;i<81;i++)state[i]=s[i]-'0';init();build();dfs(0);for(inti=0;i<RecordNumber;i++){ans[(record[i]-1)/SIZE]=(record[i]-1)%SIZE+1;}for(inti=0;i<SIZE;i++){for(intj=0;j<SIZE;j++)r[i*SIZE+j]='0'+ans[i*SIZE+j];// printf("%-2d",ans[i*SIZE+j]);//printf("\n");}r[81]='\0';return0;}intmain1(){charr[82];sudoku("070680050000000001409507000560000340000000000034000017000703804300000000090015030",r);puts(r);return0;while(input()){cnt=0;time_tstart=clock();init();build();dfs(0);output();printf("Time:%ld dfs:%d\n",clock()-start,cnt);}return0;}//////////////////////Dancing Links模版//////////////////////voidinit(void){for(inti=0;i<=MAX_C;i++){L[i]=i-1;R[i]=i+1;U[i]=i;D[i]=i;C[i]=i;O[i]=0;}L[0]=MAX_C;R[MAX_C]=0;NodeNumber=MAX_C+1;memset(S,0,sizeof(S));memset(H,0,sizeof(H));}voidinsert(inti,intj){if(H[i]){L[NodeNumber]=L[H[i]];R[NodeNumber]=H[i];L[R[NodeNumber]]=NodeNumber;R[L[NodeNumber]]=NodeNumber;}else{L[NodeNumber]=NodeNumber;R[NodeNumber]=NodeNumber;H[i]=NodeNumber;}U[NodeNumber]=U[j];D[NodeNumber]=j;U[D[NodeNumber]]=NodeNumber;D[U[NodeNumber]]=NodeNumber;C[NodeNumber]=j;O[NodeNumber]=i;S[j]++;NodeNumber++;}voidremove(intc){L[R[c]]=L[c];R[L[c]]=R[c];for(inti=D[c];i!=c;i=D[i]){for(intj=R[i];j!=i;j=R[j]){U[D[j]]=U[j];D[U[j]]=D[j];S[C[j]]--;}}}voidresume(intc){for(inti=U[c];i!=c;i=U[i]){for(intj=L[i];j!=i;j=L[j]){U[D[j]]=j;D[U[j]]=j;S[C[j]]++;}}L[R[c]]=c;R[L[c]]=c;}/* 070680050 000000001 409507000 560000340 000000000 034000017 000703804 300000000 090015030 */

编译执行,验证函数的正确性。

gcc dlx_line.cpp -o dlxline dlx_line.cpp: In function 'int main()': dlx_line.cpp:154:8: warning: ISO C++ forbids converting a string constant to 'char*' [-Wwrite-strings] 154 | sudoku("070680050000000001409507000560000340000000000034000017000703804300000000090015030",r); | ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ root@66d4e20ec1d7:/par/cppext# ./dlxline 273681459685349721419527683561978342927134568834256917152763894346892175798415236

使用官方的c++ quack插件模板程序, 在quack_extension.cpp文件中添加#include "dlx_line.cpp"#include <string>,并修改QuackScalarFun函数,其余不动。我用的是1.3版本,其他版本的函数也许不同。

...#include"dlx_line.cpp"#include<string>namespaceduckdb{inlinevoidQuackScalarFun(DataChunk&args,ExpressionState&state,Vector&result){auto&name_vector=args.data[0];UnaryExecutor::Execute<string_t,string_t>(name_vector,result,args.size(),[&](string_t name){charr[82];sudoku(name.GetString().c_str(),r);string r1=r;returnStringVector::AddString(result,"Suduku "+name.GetString()+"solved "+r1);//return StringVector::AddString(result, "Quack "+name.GetString()+" 🐥");});}

编译和转换为插件, 注意转换时插件名还沿用quack, 不要修改,否则装载插件会报错。appmeta.py是官方的python脚本

root@66d4e20ec1d7:/par/cppext# g++ -fPIC -shared -o libsudoku2.so quack_extension.cpp -I /par/include -lssl -lcrypto -I include -lduckdb -L /par/ -I /par/duckdb-0.10.3/include -lgmp root@66d4e20ec1d7:/par/cppext# python3 /par/appmeta.py -l libsudoku2.so -n sudoku -dv v1.3.0 --duckdb-platform linux_arm64 --extension-version 0.1 --abi-type "" Creating extension binary: - Input file: libsudoku2.so - Output file: sudoku.duckdb_extension - Metadata: - FIELD8 (unused) = EMPTY - FIELD7 (unused) = EMPTY - FIELD6 (unused) = EMPTY - FIELD5 (abi_type) = - FIELD4 (extension_version) = 0.1 - FIELD3 (duckdb_version) = v1.3.0 - FIELD2 (duckdb_platform) = linux_arm64 - FIELD1 (header signature) = 4 (special value to identify a duckdb extension) root@66d4e20ec1d7:/par/cppext# /par/duckdb130 -unsigned DuckDB v1.3.0 (Ossivalis) 71c5c07cdd Enter ".help" for usage hints. D load '/par/cppext/sudoku.duckdb_extension'; IO Error: Extension '/par/cppext/sudoku.duckdb_extension' did not contain the expected entrypoint function 'sudoku_init' D root@66d4e20ec1d7:/par/cppext# python3 /par/appmeta.py -l libsudoku2.so -n quack -dv v1.3.0 --duckdb-platform linux_arm64 --extension-version 0.1 --abi-type ""

执行, 可以输入字符串字面量,也可以从表中读取

DuckDB v1.3.0 (Ossivalis) 71c5c07cdd Enter ".help" for usage hints. D load '/par/cppext/quack.duckdb_extension'; D select quack('070680050000000001409507000560000340000000000034000017000703804300000000090015030'); ┌──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┐ │ quack('070680050000000001409507000560000340000000000034000017000703804300000000090015030') │ │ varchar │ ├──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤ │ Suduku 070680050000000001409507000560000340000000000034000017000703804300000000090015030solved 273681459685349721419527683561978342927134568834256917152763894346892175798415236 │ └──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┘ Run Time (s): real 0.025 user 0.052000 sys 0.000000 D select quack(replace(replace(puzzle,'?','0'),chr(10),'')) a from sudoku9_9 limit 1; ┌──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┐ │ a │ │ varchar │ ├──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤ │ Suduku 102005067684790150759621000060008205578230640291540003045900018810057020907010530solved 132485967684793152759621384463178295578239641291546873345962718816357429927814536 │ └──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┘ Run Time (s): real 0.001 user 0.004000 sys 0.000000

插件与原生c程序比较性能

D copy(select quack(p) a from read_csv('/par/1224/0114/sudoku17.txt',header=0)t(p)) to 'result17.txt'; 100% ▕████████████████████████████████████████████████████████████▏ Run Time (s): real 7.844 user 15.656000 sys 0.000000 root@66d4e20ec1d7:/par/cppext# time /par/dlxline < /par/1224/0114/sudoku17.txt > reult17.txt real 0m2.797s user 0m2.716s sys 0m0.052s

可见,尽管插件并行执行了,用时还是原生c程序的约3倍,这是因为标量函数每次执行都有切换数据库上下文的开销。4万9千行就切换了这么多次。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/3/27 6:54:59

对象库未注册-VB6企业版控件加载不了MSCOMCTL.ocx

关于WIN7下VB6中MicrosoftWindowsCommonControls6.0(SP6)加载提示“对象库未注册”的一种解决办法​​我之前在另外一台电脑上加上了进度条控件&#xff0c;使用正常&#xff1b;换了一台电脑之后&#xff0c;去“部件”中加入Microsoft Windows Common Controls 6.0 (SP6)时&a…

作者头像 李华
网站建设 2026/3/28 9:59:35

物理约束机器学习赋能科学计算

物理约束机器学习赋能科学计算 研究人员从有限体积法中汲取灵感&#xff0c;并调整神经算子&#xff0c;以在物理系统的深度学习模型中强制执行守恒定律和边界条件。 深度学习方法在科学计算领域也展现出前景&#xff0c;可用于预测偏微分方程的解。这些方程通常数值求解成本高…

作者头像 李华
网站建设 2026/3/4 23:55:16

通义千问2.5-7B-Instruct优化技巧:RTX 3060流畅运行指南

通义千问2.5-7B-Instruct优化技巧&#xff1a;RTX 3060流畅运行指南 1. 引言&#xff1a;为何在RTX 3060上部署Qwen2.5-7B-Instruct成为可能 随着大模型技术的快速演进&#xff0c;70亿参数级别的语言模型已逐步从“云端专属”走向本地化部署。通义千问2.5-7B-Instruct作为阿…

作者头像 李华
网站建设 2026/3/29 16:10:57

亲测AI印象派工坊:照片秒变艺术大作,效果超预期!

亲测AI印象派工坊&#xff1a;照片秒变艺术大作&#xff0c;效果超预期&#xff01; 关键词&#xff1a;OpenCV&#xff0c;非真实感渲染&#xff0c;图像风格迁移&#xff0c;计算摄影学&#xff0c;WebUI画廊 摘要&#xff1a;本文深入解析基于 OpenCV 计算摄影学算法构建的「…

作者头像 李华
网站建设 2026/4/2 20:08:11

VibeVoice-TTS多轮对话记忆:上下文保持能力测试案例

VibeVoice-TTS多轮对话记忆&#xff1a;上下文保持能力测试案例 1. 背景与技术挑战 在现代语音合成系统中&#xff0c;实现自然、连贯的多轮对话是一项极具挑战的任务。传统的文本转语音&#xff08;TTS&#xff09;系统通常专注于单句或短段落的语音生成&#xff0c;缺乏对上…

作者头像 李华
网站建设 2026/3/26 13:27:24

【课程设计/毕业设计】基于python卷积神经网络识别花卉是否枯萎

博主介绍&#xff1a;✌️码农一枚 &#xff0c;专注于大学生项目实战开发、讲解和毕业&#x1f6a2;文撰写修改等。全栈领域优质创作者&#xff0c;博客之星、掘金/华为云/阿里云/InfoQ等平台优质作者、专注于Java、小程序技术领域和毕业项目实战 ✌️技术范围&#xff1a;&am…

作者头像 李华