c++踩坑大法好 宏定义 头文件

//a.h
//声明一个类，和其他声明
#include <iostream>
class A{
    public:
        static int a;
        static int b;
};

//b.h
//需要a.h的类，又还有些其他声明
#include "a.h"

//main,cpp
//既需要a.h，又需要b.h，所以只能把两个头文件都引用过来。

#include "a.h"
#include "b.h"

//报错，此处重复定义了，因为一共引用了3个文件：
//1,a.h
//2,b.h
//3,a.h(b.h中引用的)
int main{
    xxx一堆代码
}

怎样修改才对？

//a.h
#ifndef hehe
#define hehe
#include <iostream>
class A{
    public:
        static int a;
        static int b;
};
#endif
//这样当第一次引入a.h的时候会声明class并且定义一个hehe变量，第二次走到
a.h的时候发现已经有了hehe了，就不会再重新声明hehe了