(多线程)leetcode1115. 交替打印FooBar 记得Thread.yield();

曾经终败给现在 2023-07-12 06:23 17阅读 0赞

我们提供一个类：

class FooBar \{  
public void foo() \{  
for (int i = 0; i < n; i++) \{  
print("foo");  
\}  
\}

public void bar() \{  
for (int i = 0; i < n; i++) \{  
print("bar");  
\}  
\}  
\}  
两个不同的线程将会共用一个 FooBar 实例。其中一个线程将会调用 foo() 方法，另一个线程将会调用 bar() 方法。

请设计修改程序，以确保 "foobar" 被输出 n 次。

示例 1:

输入: n = 1  
输出: "foobar"  
解释: 这里有两个线程被异步启动。其中一个调用 foo() 方法, 另一个调用 bar() 方法，"foobar" 将被输出一次。  
示例 2:

输入: n = 2  
输出: "foobarfoobar"  
解释: "foobar" 将被输出两次。

思路：设置一个标记即可，此标记要线程安全。

记得不符合的时候要暂时让出时间片，否则不能通过。

class FooBar {
        private int n;
        private AtomicInteger done = new AtomicInteger(0);
        public FooBar(int n) {
            this.n = n;
        }
    
        public void foo(Runnable printFoo) throws InterruptedException {
            
            for (int i = 0; i < n; i++) {
                while(done.get()!=0)Thread.yield();
            	// printFoo.run() outputs "foo". Do not change or remove this line.
            	printFoo.run();
                done.set(1);
            }
        }
    
        public void bar(Runnable printBar) throws InterruptedException {
            
            for (int i = 0; i < n; i++) {
                while(done.get()!=1)Thread.yield();
                // printBar.run() outputs "bar". Do not change or remove this line.
            	printBar.run();
                done.set(0);
            }
        }
    }