在wpf treeview中删除将跳转到父级

我尝试了这里讨论的解决方案: WPF treeview itemselected moves incorrectly when deleting an item

但是,在删除项目时,我仍然看到选择跳转到父级。我做错什么了?

主窗口

<Window x:Class="TreeViewDelete.MainWindow"
    xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
    xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
    Title="Tree View Delete sample" Height="350" Width="525">

<Window.Resources>
    <Style TargetType="TreeViewItem">
        <Setter Property="IsSelected" Value="{Binding Selected, Mode=TwoWay}"/>
        <EventSetter Event="KeyDown" Handler="OnTreeKeyDown"/>
    </Style>
    <HierarchicalDataTemplate x:Key="recursiveTemplate" ItemsSource="{Binding Children}">
        <StackPanel Orientation="Horizontal">
            <TextBlock Text="{Binding Name}"/>
        </StackPanel>
    </HierarchicalDataTemplate>
</Window.Resources>
<Grid>
    <TreeView Name="m_tree" ItemsSource="{Binding Children}" ItemTemplate="{StaticResource recursiveTemplate}"/>
</Grid>
</Window>

主窗口.xaml.cs

public partial class MainWindow : Window
{
    private Container m_root, m_child;

    public MainWindow()
    {
        InitializeComponent();

        m_root = new Container("Root");

        m_child = new Container("main");

        m_child.Add(new Container("k1"));
        m_child.Add(new Container("k2"));
        m_child.Add(new Container("k3"));
        m_child.Add(new Container("k4"));
        m_child.Add(new Container("k5"));

        m_root.Add(m_child);

        m_tree.DataContext = m_root;
    }

    private IEnumerable<T> GetVisualAncestorsOfType<T>(DependencyObject obj) where T : DependencyObject
    {
        for (; obj != null; obj = VisualTreeHelper.GetParent(obj))
            if (obj is T)
                yield return (T)obj;
    }

    private void OnTreeKeyDown(object sender, KeyEventArgs e)
    {
        switch (e.Key)
        {
            case Key.Delete:
                {
                    TreeViewItem item = sender as TreeViewItem;

                    if (item != null)
                    {
                        Container container = item.Header as Container;

                        if (container != null)
                        {
                            Container parent = container.Parent;

                            // Find the currently focused element in the TreeView's focus scope
                            DependencyObject focused =
                              FocusManager.GetFocusedElement(
                                FocusManager.GetFocusScope(m_tree)) as DependencyObject;

                            // Scan up the VisualTree to find the TreeViewItem for the parent
                            var parentContainer = (
                              from element in GetVisualAncestorsOfType<FrameworkElement>(focused)
                              where (element is TreeViewItem && element.DataContext == parent)
                                    || element is TreeView
                              select element
                              ).FirstOrDefault();

                            parent.Remove(container);

                            if (parentContainer != null)
                            {
                                parent.Children[0].Selected = true;
                                parentContainer.Focus();
                            }
                        }
                    }

                    e.Handled = true;
                }
                break;
        }
    }
}

最后是container.cs

public class Container : INotifyPropertyChanged
{
    private bool m_selected;
    private string m_name;
    private ObservableCollection<Container> m_children;
    private Container m_parent;

    public Container(string name)
    {
        m_name = name;
        m_children = new ObservableCollection<Container>();
        m_parent = null;
        m_selected = false;
    }

    public event PropertyChangedEventHandler PropertyChanged;

    public string Name
    {
        get
        {
            return m_name;
        }

        set
        {
            m_name = value;

            OnPropertyChanged("Name");
        }
    }

    public Container Parent
    {
        get
        {
            return m_parent;
        }

        set
        {
            m_parent = value;
        }
    }

    public bool Selected
    {
        get
        {
            return m_selected;
        }

        set
        {
            m_selected = value;

            OnPropertyChanged("Selected");
        }
    }

    public ObservableCollection<Container> Children
    {
        get
        {
            return m_children;
        }
    }

    private void OnPropertyChanged(string name)
    {
        if (PropertyChanged != null)
        {
            PropertyChanged(this, new PropertyChangedEventArgs(name));
        }
    }

    public void Add(Container child)
    {
        m_children.Add(child);

        child.Parent = this;
    }

    public void Remove(Container child)
    {
        m_children.Remove(child);

        child.Parent = null;
    }
}