股东或高管以借款形式占用公司资金，公司应采取以下措施应对：

1. 完善内控机制：建立严格的借款审批流程，确保所有借款行为符合公司章程和法律法规，防止未经授权的借款发生。

2. 透明披露：根据《企业会计准则第41号——在其他主体中权益的披露(2014)》，如果合营企业或联营企业资金转移能力受限，公司应披露相关信息，包括限制的性质和程度。

3. 合规会计处理：根据《企业会计准则第40号——合营安排(2014)》，对合营企业具有重大影响的，应按照长期股权投资准则处理；不具有重大影响的，按照金融工具确认和计量准则处理。

4. 监控资金流动：定期审计和监控资金流向，确保资金使用符合公司利益，防止资金被滥用。

5. 法律约束：在借款协议中明确还款期限、利率和违约责任，必要时可寻求法律保障，如设定担保。

6. 强化治理：确保董事会和监事会有效运作，防止控股股东或高管滥用权力。

7. 风险管理：评估此类借款对财务报表的影响，包括信用风险和流动性风险，并在财务报告中充分披露。

8. 合规审计：会计师事务所在审计时应严格审查此类交易，确保其合规性，避免审计风险。

通过以上措施，公司可以降低股东或高管占用资金带来的风险，保护公司和所有股东的权益。