在ReactJS中状态发生变化后，有没有办法立即重置状态？

constructor(props) {
  super(props);
  this.state = {
    // 初始状态
    count: 0
  };
}

resetState() {
  // 重置状态为初始值
  this.setState({ count: 0 });
}

class MyComponent extends React.Component {
  constructor(props) {
    super(props);
    this.state = {
      count: 0
    };
  }

  resetState = () => {
    // 重置状态为初始值
    this.setState({ count: 0 });
  }

  render() {
    return (
      <div>
        <p>Count: {this.state.count}</p>
        <button onClick={() => this.setState({ count: this.state.count + 1 })}>
          Increment
        </button>
        <button onClick={this.resetState}>
          Reset
        </button>
      </div>
    );
  }
}

import React, { useState } from 'react';

function MyComponent() {
  const [count, setCount] = useState(0);

  const resetState = () => {
    // 重置状态为初始值
    setCount(0);
  }

  return (
    <div>
      <p>Count: {count}</p>
      <button onClick={() => setCount(count + 1)}>
        Increment
      </button>
      <button onClick={resetState}>
        Reset
      </button>
    </div>
  );
}