If you root your device, can you keep everything the same and just get the latest updates like a nexus?

I have a note 2 from att and would love to receive the latest updates from google but Ive heard doing so will cause me to lose my S pen functionality which I use very often. I am also considering getting a note 8 tablet but I am beginning to worry since it seems these devices only get a handful of updates and then support stops for them. If I root these devices can I keep touchwiz and s pen functionality while receiving the latest firmware updates?